diff --git a/agent/component/__init__.py b/agent/component/__init__.py
index ea9e745bef8..199015e47f5 100644
--- a/agent/component/__init__.py
+++ b/agent/component/__init__.py
@@ -20,6 +20,6 @@
 
 
 def component_class(class_name):
-    m = importlib.import_module("graph.component")
+    m = importlib.import_module("agent.component")
     c = getattr(m, class_name)
     return c