diff --git a/tests/basic/test_commands.py b/tests/basic/test_commands.py
index 84d152127..286bd30ed 100644
--- a/tests/basic/test_commands.py
+++ b/tests/basic/test_commands.py
@@ -1309,7 +1309,8 @@ class TestCommands(TestCase):
                 # Check that the tool output shows the correct value with format
                 # Use the actual input_value (not normalized) in the assertion
                 mock_tool_output.assert_any_call(
-                    f"Set thinking token budget to {expected_tokens:,} tokens ({input_value.lower()})."
+                    f"Set thinking token budget to {expected_tokens:,} tokens"
+                    f" ({input_value.lower()})."
                 )
 
         # Test with no value provided - should display current value