diff --git a/mteb/models/voyage_models.py b/mteb/models/voyage_models.py
index 1c628dda13..539f3b5c38 100644
--- a/mteb/models/voyage_models.py
+++ b/mteb/models/voyage_models.py
@@ -99,7 +99,8 @@ def encode(
         **kwargs: Any,
     ) -> np.ndarray:
         prompt_name = self.get_prompt_name(self.model_prompts, task_name, prompt_type)
-        input_type = prompt_name if prompt_name is not None else "document"
+        input_type = self.model_prompts.get(prompt_name, "document")
+
         return self._batched_encode(sentences, batch_size, input_type)
 
     def _batched_encode(