diff --git a/projects/orquestra-sdk/tests/cli/test_arg_resolvers.py b/projects/orquestra-sdk/tests/cli/test_arg_resolvers.py
index 00ac648be..b26d675cc 100644
--- a/projects/orquestra-sdk/tests/cli/test_arg_resolvers.py
+++ b/projects/orquestra-sdk/tests/cli/test_arg_resolvers.py
@@ -885,6 +885,7 @@ def test_choices_limited_by_availibility(
             if other_choice:
                 expected_choices.append(WorkflowLogs.WorkflowLogTypeName.OTHER)
 
+            context: t.Union[contextlib.nullcontext, pytest.WarningsRecorder]
             if not any([task_choice, system_choice, env_choice, other_choice]):
                 expected_call = 0
                 context = pytest.warns(UserWarning)