diff --git a/tests/chat/test_chat_app.py b/tests/chat/test_chat_app.py
index 66e7f7e4..501e527a 100644
--- a/tests/chat/test_chat_app.py
+++ b/tests/chat/test_chat_app.py
@@ -4,7 +4,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
 import pytest
 from typer.testing import CliRunner
 
-from haiku.rag.cli import cli
+from haiku.rag.cli import _cli as cli
 
 runner = CliRunner()
 
diff --git a/tests/test_inspector.py b/tests/test_inspector.py
index 92a6005e..1f9c82b4 100644
--- a/tests/test_inspector.py
+++ b/tests/test_inspector.py
@@ -3,7 +3,7 @@ from unittest.mock import AsyncMock, patch
 import pytest
 from typer.testing import CliRunner
 
-from haiku.rag.cli import cli
+from haiku.rag.cli import _cli as cli
 from haiku.rag.store.models import Document
 
 runner = CliRunner()