diff --git a/connpy/ai.py b/connpy/ai.py
index 3df4df7..05ac08c 100755
--- a/connpy/ai.py
+++ b/connpy/ai.py
@@ -156,7 +156,7 @@ Users will typically use words like verify, check, make sure, or similar to refe
             try:
                 myfunction = function(*args)
                 break
-            except (openai.error.APIConnectionError, openai.error.RateLimitError):
+            except:
                 wait_time = backoff_num * (2 ** retries)
                 time.sleep(wait_time)
                 retries += 1