diff --git a/scrapegraphai/utils/parse_state_keys.py b/scrapegraphai/utils/parse_state_keys.py
index 6afc2ecb..85712ef6 100644
--- a/scrapegraphai/utils/parse_state_keys.py
+++ b/scrapegraphai/utils/parse_state_keys.py
@@ -101,18 +101,3 @@ def parse_expression(expression, state: dict) -> list:
             final_result.append(key)
 
     return final_result
-
-
-EXPRESSION = "user_input & (relevant_chunks | parsed_document | document)"
-state = {
-    "user_input": None,
-    "document": None,
-    "parsed_document": None,
-    "relevant_chunks": None,
-}
-
-try:
-    result = parse_expression(EXPRESSION, state)
-    print("Matched keys:", result)
-except ValueError as e:
-    print("Error:", e)
diff --git a/tests/utils/parse_state_keys_test.py b/tests/utils/parse_state_keys_test.py
new file mode 100644
index 00000000..d91355f1
--- /dev/null
+++ b/tests/utils/parse_state_keys_test.py
@@ -0,0 +1,21 @@
+"""
+Parse_state_key test module 
+"""
+import pytest
+from scrapegraphai.utils.parse_state_keys import parse_expression
+
+
+def test_parse_expression():
+    """Test parse_expression function."""
+    EXPRESSION = "user_input & (relevant_chunks | parsed_document | document)"
+    state = {
+        "user_input": None,
+        "document": None,
+        "parsed_document": None,
+        "relevant_chunks": None,
+    }
+    try:
+        result = parse_expression(EXPRESSION, state)
+        assert result != []
+    except ValueError as e:
+        assert "Error" in str(e)