diff --git a/cognee/tests/test_library.py b/cognee/tests/test_library.py
index e45807836..3554a7626 100755
--- a/cognee/tests/test_library.py
+++ b/cognee/tests/test_library.py
@@ -17,9 +17,9 @@ async def  main():
     explanation_file_path = path.join("/Users/runner/work/cognee/cognee/./cognee/tests", "test_data/Natural_language_processing.txt")
     await cognee.add([explanation_file_path], dataset_name)
     
-    dataset_name = "short_stories"
-    # data_directory_path is defined above
-    await cognee.add("data://" + "/Users/runner/work/cognee/cognee/./cognee/tests", dataset_name)
+    # dataset_name = "short_stories"
+    # # data_directory_path is defined above
+    # await cognee.add("data://" + "/Users/runner/work/cognee/cognee/./cognee/tests", dataset_name)
 
     text_1 = """A quantum computer is a computer that takes advantage of quantum mechanical phenomena.
     At small scales, physical matter exhibits properties of both particles and waves, and quantum computing leverages this behavior, specifically quantum superposition and entanglement, using specialized hardware that supports the preparation and manipulation of quantum states.
@@ -45,7 +45,7 @@ async def  main():
         dataset_name
     )
 
-    await cognee.cognify(["short_stories", "cs_explanations"])
+    await cognee.cognify([ "cs_explanations"])
 
 
 if __name__ == "__main__":