diff --git a/autogpt/api_manager.py b/autogpt/api_manager.py
index 52b94ff7..b16f3af7 100644
--- a/autogpt/api_manager.py
+++ b/autogpt/api_manager.py
@@ -87,7 +87,7 @@ class ApiManager(metaclass=Singleton):
         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