diff --git a/legacy/app/modules/intelligence/tools/kg_based_tools/get_code_from_node_id_tool.py b/legacy/app/modules/intelligence/tools/kg_based_tools/get_code_from_node_id_tool.py
index efbb49e48..e36dbb5e9 100644
--- a/legacy/app/modules/intelligence/tools/kg_based_tools/get_code_from_node_id_tool.py
+++ b/legacy/app/modules/intelligence/tools/kg_based_tools/get_code_from_node_id_tool.py
@@ -219,7 +219,8 @@ def _process_result(
                 node_id=node_id,
                 project_id=project.id,
             )
-        return truncated_result
+            return truncated_result
+        return result
 
     @staticmethod
     def _get_relative_file_path(file_path: str) -> str: