diff --git a/tests/test_cli_context_warning.py b/tests/test_cli_context_warning.py
index abf9c1349..bf0c5aac4 100644
--- a/tests/test_cli_context_warning.py
+++ b/tests/test_cli_context_warning.py
@@ -32,6 +32,8 @@ def cli_obj(_isolate):
         obj.session_id = None
         obj.api_key = "test"
         obj.base_url = ""
+        obj.provider = "test"
+        obj._provider_source = None
         # Mock agent with context compressor
         obj.agent = SimpleNamespace(
             context_compressor=SimpleNamespace(context_length=None)