update the modules to work with the callbacks