diff --git a/cognee/tests/cli_tests/cli_integration_tests/test_cli_integration.py b/cognee/tests/cli_tests/cli_integration_tests/test_cli_integration.py
index 67ebca926..9b350bb7e 100644
--- a/cognee/tests/cli_tests/cli_integration_tests/test_cli_integration.py
+++ b/cognee/tests/cli_tests/cli_integration_tests/test_cli_integration.py
@@ -103,12 +103,7 @@ class TestCliIntegration:
             )
 
             try:
-                assert (
-                    not has_error
-                    or has_expected_failure
-                    or has_litellm_cancellation
-                    or has_kuzu_extension_error
-                )
+                assert not has_error or has_expected_failure or has_litellm_cancellation
             except Exception as e:
                 logger.error(f"{str(e)}", exc_info=True)
                 import pprint