diff --git a/bot/openai/open_ai_image.py b/bot/openai/open_ai_image.py
index 974bf82..c80650f 100644
--- a/bot/openai/open_ai_image.py
+++ b/bot/openai/open_ai_image.py
@@ -12,6 +12,7 @@ from config import conf
 class OpenAIImage(object):
     def __init__(self):
         openai.api_key = conf().get("open_ai_api_key")
+        openai.api_base = conf().get("open_ai_api_base")
         if conf().get("rate_limit_dalle"):
             self.tb4dalle = TokenBucket(conf().get("rate_limit_dalle", 50))