diff --git a/main.py b/main.py
index e277d3a..bb1fa8d 100644
--- a/main.py
+++ b/main.py
@@ -152,6 +152,7 @@ async def load_module_from_path(module_name, module_path):
         # Load the module
         spec = importlib.util.spec_from_file_location(module_name, module_path)
         module = importlib.util.module_from_spec(spec)
+        sys.modules[module_name] = module
         spec.loader.exec_module(module)
         print(f"Loaded module: {module.__name__}")
         if hasattr(module, "Pipeline"):