diff --git a/docs/building-with-codegen/traversing-the-call-graph.mdx b/docs/building-with-codegen/traversing-the-call-graph.mdx
index 43336bb3d..27acc5b9c 100644
--- a/docs/building-with-codegen/traversing-the-call-graph.mdx
+++ b/docs/building-with-codegen/traversing-the-call-graph.mdx
@@ -47,6 +47,10 @@ def create_call_graph(start_func, end_func, max_depth=5):
             src_func = parent_func.function_definition
             src_call = parent_func
 
+        # Skip external modules
+        if isinstance(src_func, ExternalModule):
+            return
+
         # Traverse all function calls
         for call in src_func.function_calls:
             func = call.function_definition