diff --git a/tests/__init__.py b/tests/__init__.py
index 0fd6274..0d88f0b 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -47,8 +47,10 @@ def _import_from(mod, path, mod_dir=None):
             mod_info = imp.find_module(mod_dir, [path])
             return imp.load_module(mod, *mod_info)
         else:
-            mod_info = importlib.machinery.PathFinder().find_spec(mod_dir, [path])
-            return importlib.import_module(mod, *mod_info)
+            spec = importlib.machinery.PathFinder().find_spec(mod_dir, [path])
+            module = importlib.util.module_from_spec(spec)
+            sys.modules[mod] = module
+            spec.loader.exec_module(module)
     except ImportError:
         return None