diff --git a/src/text_generation/adapters/foundation_models/apple_openelm_foundation_model.py b/src/text_generation/adapters/foundation_models/apple_openelm_foundation_model.py
index c51ad636b..dbfee306f 100644
--- a/src/text_generation/adapters/foundation_models/apple_openelm_foundation_model.py
+++ b/src/text_generation/adapters/foundation_models/apple_openelm_foundation_model.py
@@ -8,7 +8,7 @@ from src.text_generation.common.model_id import ModelId
 class AppleOpenELMFoundationModel(BaseFoundationModel):
     """apple/OpenELM-3B-Instruct implementation"""
 
-    MODEL_ID = ModelId.APPLE_OPENELM_3B_INSTRUCT.value
+    MODEL_ID = ModelId.APPLE_OPENELM_3B_INSTRUCT
     
     def __init__(self, config: AppleOpenELMConfig = AppleOpenELMConfig()):
         self.config = config
diff --git a/src/text_generation/adapters/foundation_models/meta_llama_foundation_model.py b/src/text_generation/adapters/foundation_models/meta_llama_foundation_model.py
index 199842c4e..beedc12ae 100644
--- a/src/text_generation/adapters/foundation_models/meta_llama_foundation_model.py
+++ b/src/text_generation/adapters/foundation_models/meta_llama_foundation_model.py
@@ -1,17 +1,14 @@
-from src.text_generation.adapters.foundation_models.config.meta_llama_config import MetaLlamaConfig
-from src.text_generation.adapters.foundation_models.base.base_foundation_model import BaseFoundationModel
-
-
 from langchain_huggingface import HuggingFacePipeline
 from transformers import AutoModelForCausalLM, AutoTokenizer
-
+from src.text_generation.adapters.foundation_models.config.meta_llama_config import MetaLlamaConfig
+from src.text_generation.adapters.foundation_models.base.base_foundation_model import BaseFoundationModel
 from src.text_generation.common.model_id import ModelId
 
 
 class MetaLlamaFoundationModel(BaseFoundationModel):
     """meta-llama/Llama-3.2-3B-Instruct implementation"""
 
-    MODEL_ID = ModelId.META_LLAMA_3_2_3B_INSTRUCT.value
+    MODEL_ID = ModelId.META_LLAMA_3_2_3B_INSTRUCT
     
     def __init__(self, config: MetaLlamaConfig = MetaLlamaConfig()):
         self.config = config
diff --git a/src/text_generation/adapters/foundation_models/microsoft_phi3_foundation_model.py b/src/text_generation/adapters/foundation_models/microsoft_phi3_foundation_model.py
index 68e8726d8..25451a6fb 100644
--- a/src/text_generation/adapters/foundation_models/microsoft_phi3_foundation_model.py
+++ b/src/text_generation/adapters/foundation_models/microsoft_phi3_foundation_model.py
@@ -1,5 +1,4 @@
 from langchain_huggingface import HuggingFacePipeline
-from optimum.onnxruntime import ORTModelForCausalLM
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from src.text_generation.adapters.foundation_models.config.microsoft_phi3mini4k_config import MicrosoftPhi3Mini4KConfig
 from src.text_generation.adapters.foundation_models.base.base_foundation_model import BaseFoundationModel