diff --git a/bot/openai_helper.py b/bot/openai_helper.py
index 26fad44..9065d9d 100644
--- a/bot/openai_helper.py
+++ b/bot/openai_helper.py
@@ -34,12 +34,12 @@ def default_max_tokens(model: str) -> int:
     """
     base = 1200
     if model in GPT_3_MODELS:
-        if model == "gpt-3.5-turbo-0301":
-            return 4096
         return base
     elif model in GPT_4_MODELS:
         return base * 2
-    elif model in GPT_3_16K_MODELS:
+    elif model in GPT_3_16K_MODELS:    
+        if model == "gpt-3.5-turbo-0301":
+            return 4096
         return base * 4
     elif model in GPT_4_32K_MODELS:
         return base * 8