diff --git a/scripts/encode_data.py b/scripts/encode_data.py
index 6a82e28..94e1c8e 100644
--- a/scripts/encode_data.py
+++ b/scripts/encode_data.py
@@ -76,7 +76,7 @@ def _generate_child_docs(self, parent_document):
                 "vector": doc[0],
                 "document": doc[1]
             })
-        return child_docs
+        return document_map
 
     
     def _process_document(self, file) -> None: