diff --git a/autointent/_embedder.py b/autointent/_embedder.py
index 81a6b94c4..4de8f46fa 100644
--- a/autointent/_embedder.py
+++ b/autointent/_embedder.py
@@ -178,7 +178,7 @@ def embed(self, utterances: list[str], task_type: TaskTypeEnum | None = None) ->
             convert_to_numpy=True,
             batch_size=self.config.batch_size,
             normalize_embeddings=True,
-            prompt_name=self.config.get_prompt_type(task_type),
+            prompt=self.config.get_prompt_type(task_type),
         )
 
         if self.config.use_cache: