Upgrading 2 Max A40630 to a A40640 module

I have a 2 Max 30w and I want to get the new A40640 module that has the air assist on it. Can I just swap the modules out or do I have to replace the driver board? Thanks

Can be replaced directly, no need to replace the main board of the machine