diff --git a/prime_prompt/generate_prime_promt.py b/prime_prompt/generate_prime_promt.py
index b0ab099..a8c736d 100644
--- a/prime_prompt/generate_prime_promt.py
+++ b/prime_prompt/generate_prime_promt.py
@@ -29,7 +29,7 @@ def scan_docs(path: str):
                     break
                 function_documentation += lines[i]
                 i += 1
-            yield function_name, documentation
+            yield function_name, function_documentation
 
 
 llm_related_modules = ["data_collection"]