diff --git a/cogs/prompt_optimizer_cog.py b/cogs/prompt_optimizer_cog.py
index efd6f2c..715db69 100644
--- a/cogs/prompt_optimizer_cog.py
+++ b/cogs/prompt_optimizer_cog.py
@@ -96,7 +96,7 @@ class ImgPromptOptimizer(discord.Cog, name="ImgPromptOptimizer"):
                 best_of_override=1,
                 max_tokens_override=60,
                 custom_api_key=user_api_key,
-                is_chatgpt_request="turbo" in str(self.model.model),
+                is_chatgpt_request="turbo" in str(self.model.model) or "gpt-4" in str(self.model.model),
             )
 
             # THIS USES MORE TOKENS THAN A NORMAL REQUEST! This will use roughly 4000 tokens, and will repeat the query
@@ -105,7 +105,7 @@ class ImgPromptOptimizer(discord.Cog, name="ImgPromptOptimizer"):
 
             response_text = (
                 str(response["choices"][0]["text"])
-                if not self.model.model in Models.CHATGPT_MODELS
+                if not (self.model.model in Models.CHATGPT_MODELS or self.model.model in Models.GPT4_MODELS)
                 else response["choices"][0]["message"]["content"]
             )