diff --git a/biochatter/prompts.py b/biochatter/prompts.py
index b7bd1132..a544013e 100644
--- a/biochatter/prompts.py
+++ b/biochatter/prompts.py
@@ -193,7 +193,7 @@ def _select_entities(self, question: str) -> bool:
         self.question = question
 
         conversation = GptConversation(
-            model_name="gpt-3.5-turbo",
+            model_name=self.model_name,
             prompts={},
             correct=False,
         )
@@ -498,7 +498,7 @@ def _generate_query(
             for key, value in self.rel_directions.items():
                 for pair in value:
                     msg += f"'(:{pair[0]})-(:{key})->(:{pair[1]})', "
-            msg += "generate a Cypher query using one of these combinations. "
+            msg += f"generate a {query_language} query using one of these combinations. "
 
         msg += "Only return the query, without any additional text."