Fix issue with import_module