diff --git a/cognee/tests/test_library.py b/cognee/tests/test_library.py
index 82c14d40f..949df5282 100755
--- a/cognee/tests/test_library.py
+++ b/cognee/tests/test_library.py
@@ -3,14 +3,14 @@ async def  main():
     import pathlib
     import cognee
 
-    # print("Working dir: ", str(pathlib.Path(__file__).parent))
-    # data_directory_path = str(pathlib.Path(path.join(pathlib.Path(__file__).parent, "../../.data")).resolve())
-    # print("Data dir: ", data_directory_path)
-    # cognee.config.data_root_directory(data_directory_path)
-    #
-    # cognee_directory_path = str(pathlib.Path(path.join(pathlib.Path(__file__).parent, "../../.cognee_system")).resolve())
-    # print("System dir: ", cognee_directory_path)
-    # cognee.config.system_root_directory(cognee_directory_path)
+    print("Working dir: ", str(pathlib.Path(__file__).parent))
+    data_directory_path = str(pathlib.Path(path.join(pathlib.Path(__file__).parent, "../../.data")).resolve())
+    print("Data dir: ", data_directory_path)
+    cognee.config.data_root_directory(data_directory_path)
+
+    cognee_directory_path = str(pathlib.Path(path.join(pathlib.Path(__file__).parent, "../../.cognee_system")).resolve())
+    print("System dir: ", cognee_directory_path)
+    cognee.config.system_root_directory(cognee_directory_path)
 
 
     dataset_name = "cs_explanations"