diff --git a/api/services/extension_process.py b/api/services/extension_process.py
index d063fc7..df0e8f7 100644
--- a/api/services/extension_process.py
+++ b/api/services/extension_process.py
@@ -257,7 +257,7 @@ def load(self) -> None:
         self._send({"action": "load"})
 
         msg = self._recv(timeout=None)  # model load can be arbitrarily slow
-        if msg.get("type") == "loaded":
+        if msg.get('type') in ['loaded', 'ready']:
             self._loaded = True
         elif msg.get("type") == "error":
             raise RuntimeError(msg.get("traceback") or msg.get("message"))