Temporary workaround for import_all_modules

This should be fixed asap.