diff --git a/task_manager/byllm/BE/v2/agent_core.jac b/task_manager/byllm/BE/v2/agent_core.jac
index 0a991498..c47e4986 100644
--- a/task_manager/byllm/BE/v2/agent_core.jac
+++ b/task_manager/byllm/BE/v2/agent_core.jac
@@ -43,7 +43,7 @@ walker agent {
         static has auth: bool = False;
     }
     def get_node_class(class_name: str) {
-        main_mod = sys.modules.get("__main__");
+        main_mod = sys.modules.get("main");
         if main_mod and hasattr(main_mod, class_name) {
             return getattr(main_mod, class_name, None);
         }
@@ -76,7 +76,7 @@ walker agent {
         }
         node_inst = node_cls();
         visit [-->(`?node_cls)] else {
-            attached_routed_node = here ++> node_inst;
+            attached_routed_node = self.session ++> node_inst;
             visit attached_routed_node;
         }
     }
@@ -103,4 +103,4 @@ walker get_all_sessions {
 
 with entry {
     load_dotenv();
-}
\ No newline at end of file
+}