diff --git a/autogpt/api_manager.py b/autogpt/api_manager.py
index ace64fbdaf32..d69ffbf684d5 100644
--- a/autogpt/api_manager.py
+++ b/autogpt/api_manager.py
@@ -89,7 +89,7 @@ def set_total_budget(self, total_budget):
         Sets the total user-defined budget for API calls.
 
         Args:
-        prompt_tokens (int): The number of tokens used in the prompt.
+        total_budget (float): The total budget for API calls.
         """
         self.total_budget = total_budget