diff --git a/mteb/models/gritlm_models.py b/mteb/models/gritlm_models.py
index 1e9c85eb83..5e52fe4c50 100644
--- a/mteb/models/gritlm_models.py
+++ b/mteb/models/gritlm_models.py
@@ -10,7 +10,7 @@
 logger = logging.getLogger(__name__)
 
 
-def gritlm_instruction(instruction: str = "") -> str:
+def gritlm_instruction(instruction: str = "", prompt_type=None) -> str:
     return (
         "<|user|>\n" + instruction + "\n<|embed|>\n" if instruction else "<|embed|>\n"
     )