diff --git a/blendsql/models/remote/_openai.py b/blendsql/models/remote/_openai.py
index b8911cb2..78a35d7c 100644
--- a/blendsql/models/remote/_openai.py
+++ b/blendsql/models/remote/_openai.py
@@ -75,7 +75,8 @@ def _load_model(self) -> Model:
         )
 
     def _setup(self, **kwargs) -> None:
-        return openai_setup()
+        openai_setup()
+        self.model = self._load_model()
 
 
 class OpenaiLLM(Model):
@@ -103,4 +104,5 @@ def _load_model(self) -> Model:
         )
 
     def _setup(self, **kwargs) -> None:
-        return openai_setup()
+        openai_setup()
+        self.model = self._load_model()