diff --git a/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/__init__.py b/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/__init__.py
index 2054b2be..0dbe53c1 100644
--- a/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/__init__.py
+++ b/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/__init__.py
@@ -2,9 +2,10 @@
 from pydantic import Field
 from typing import Literal
 from module_text_llm.retrieval_augmented_generation.agents import TutorAgent
-tutor = TutorAgent()
 from module_text_llm.retrieval_augmented_generation.prompt_generate_suggestions import GenerateSuggestionsPrompt
 
+tutor = TutorAgent()
+
 class RAGApproachConfig(ApproachConfig):
     type: Literal['rag'] = 'rag'
     generate_suggestions_prompt: GenerateSuggestionsPrompt = Field(default=GenerateSuggestionsPrompt())
diff --git a/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/generate_suggestions.py b/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/generate_suggestions.py
index 346b5e37..521eacdd 100644
--- a/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/generate_suggestions.py
+++ b/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/generate_suggestions.py
@@ -8,7 +8,6 @@
     check_prompt_length_and_omit_features_if_necessary, 
     num_tokens_from_prompt,
 )
-from athena.text import Exercise, Submission, Feedback
 from module_text_llm.config import BasicApproachConfig
 from module_text_llm.helpers.utils import add_sentence_numbers, get_index_range_from_line_range, format_grading_instructions
 from module_text_llm.basic_approach.prompt_generate_suggestions import AssessmentModel
diff --git a/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/prompt_generate_suggestions.py b/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/prompt_generate_suggestions.py
index 6599fc7d..7988b9b4 100644
--- a/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/prompt_generate_suggestions.py
+++ b/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/prompt_generate_suggestions.py
@@ -1,6 +1,5 @@
 from pydantic import Field, BaseModel
 from typing import List, Optional
-from pydantic import BaseModel, Field
 
 system_message = """\
 You are an AI tutor for text assessment at a prestigious university.