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 9b350bb7e..8c4268dbc 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
@@ -82,6 +82,7 @@ class TestCliIntegration:
                 capture_output=True,
                 text=True,
                 cwd=Path(__file__).parent.parent.parent,  # Go to project root
+                check=True,
             )
 
             # Note: This might fail due to dependencies, but we're testing the CLI structure
@@ -94,13 +95,6 @@ class TestCliIntegration:
             has_litellm_cancellation = (
                 "loggingworker cancelled" in stderr_lower or "cancellederror" in stderr_lower
             )
-            has_kuzu_extension_error = (
-                "could not establish connection" in stderr_lower
-                and "extension.kuzudb.com" in stderr_lower
-            ) or (
-                "binder exception: extension" in stderr_lower
-                and "has not been installed" in stderr_lower
-            )
 
             try:
                 assert not has_error or has_expected_failure or has_litellm_cancellation