diff --git a/aider/commands.py b/aider/commands.py
index 1232bf057..56295c02c 100644
--- a/aider/commands.py
+++ b/aider/commands.py
@@ -1421,12 +1421,8 @@ class Commands:
         model = self.coder.main_model
         model.set_thinking_tokens(value)
 
-        # Try to display the actual value that was set
-        if model.extra_params and "thinking" in model.extra_params:
-            budget = model.extra_params["thinking"].get("budget_tokens")
-            self.io.tool_output(f"Set thinking token budget to {budget:,} tokens.")
-        else:
-            self.io.tool_output(f"Set thinking token budget to {value}.")
+        budget = model.extra_params["thinking"].get("budget_tokens")
+        self.io.tool_output(f"Set thinking token budget to {budget:,} tokens.")
 
     def cmd_copy_context(self, args=None):
         """Copy the current chat context as markdown, suitable to paste into a web UI"""