diff --git a/litellm/budget_manager.py b/litellm/budget_manager.py
index bb7b5dff9..473bb93e2 100644
--- a/litellm/budget_manager.py
+++ b/litellm/budget_manager.py
@@ -41,7 +41,7 @@ class BudgetManager:
             else:
                 self.user_dict = response["data"]
 
-    def create_budget(self, total_budget: float, user: str, duration: Literal["daily", "weekly", "monthly", "yearly"] = None, created_at: float = time.time()): 
+    def create_budget(self, total_budget: float, user: str, duration: Optional[Literal["daily", "weekly", "monthly", "yearly"]] = None, created_at: float = time.time()): 
         self.user_dict[user] = {"total_budget": total_budget}
         if duration is None:
             return self.user_dict[user]
diff --git a/pyproject.toml b/pyproject.toml
index 0c26a5e66..9f73e68f1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "litellm"
-version = "0.1.608"
+version = "0.1.609"
 description = "Library to easily interface with LLM API providers"
 authors = ["BerriAI"]
 license = "MIT License"