diff --git a/bondai/models/openai/openai_embedding_model.py b/bondai/models/openai/openai_embedding_model.py
index 0b061e5..581ca5d 100644
--- a/bondai/models/openai/openai_embedding_model.py
+++ b/bondai/models/openai/openai_embedding_model.py
@@ -11,7 +11,7 @@ def __init__(
         model: OpenAIModelNames = OpenAIModelNames.TEXT_EMBEDDING_ADA_002,
         connection_params: Dict = None,
     ):
-        self._model = model.value
+        self._model = model.value if isinstance(model, OpenAIModelNames) else model
         self._connection_params = connection_params
         if ModelConfig[self._model]["model_type"] != OpenAIModelType.EMBEDDING:
             raise Exception(f"Model {model} is not an embedding model.")