diff --git a/prompt2model/utils/api_tools.py b/prompt2model/utils/api_tools.py
index 6a4c1d8b6..139f5c24f 100644
--- a/prompt2model/utils/api_tools.py
+++ b/prompt2model/utils/api_tools.py
@@ -169,14 +169,14 @@ async def _throttled_completion_acreate(
                         if isinstance(
                             e,
                             (
-                                openai.ServiceUnavailableError,
+                                openai.APIStatusError,
                                 openai.APIError,
                             ),
                         ):
                             logging.warning(
                                 ERROR_ERRORS_TO_MESSAGES[type(e)].format(e=e)
                             )
-                        elif isinstance(e, openai.InvalidRequestError):
+                        elif isinstance(e, openai.BadRequestError):
                             logging.warning(ERROR_ERRORS_TO_MESSAGES[type(e)])
                             return {
                                 "choices": [