diff --git a/modules/modeling/module_modeling_llm/module_modeling_llm/generate_suggestions.py b/modules/modeling/module_modeling_llm/module_modeling_llm/generate_suggestions.py
index 1e24e0fc..f3781d3e 100644
--- a/modules/modeling/module_modeling_llm/module_modeling_llm/generate_suggestions.py
+++ b/modules/modeling/module_modeling_llm/module_modeling_llm/generate_suggestions.py
@@ -5,7 +5,6 @@
 from pydantic import BaseModel, Field
 
 from athena import emit_meta
-from athena.logger import logger
 from athena.modeling import Exercise, Submission, Feedback
 from module_modeling_llm.config import BasicApproachConfig
 from langchain_core.output_parsers import PydanticOutputParser
diff --git a/modules/modeling/module_modeling_llm/module_modeling_llm/helpers/llm_utils.py b/modules/modeling/module_modeling_llm/module_modeling_llm/helpers/llm_utils.py
index b5194d08..9c136027 100644
--- a/modules/modeling/module_modeling_llm/module_modeling_llm/helpers/llm_utils.py
+++ b/modules/modeling/module_modeling_llm/module_modeling_llm/helpers/llm_utils.py
@@ -4,7 +4,7 @@
 from langchain_core.pydantic_v1 import BaseModel, ValidationError
 from langchain_core.runnables import RunnableSequence
 from athena import get_experiment_environment
-from assessment_module_manager.logger import logger
+from athena.logger import logger
 
 T = TypeVar("T", bound=BaseModel)
 
@@ -46,7 +46,7 @@ async def predict_and_parse(
     try:
         return await chain.ainvoke(prompt_input, config={"tags": tags})
     except (ValidationError) as e:
-        
+
         logger.error("Exception type: %s, Message: %s", type(e).__name__, e)
 
         return None
\ No newline at end of file