diff --git a/kazu/steps/ner/llm_ner.py b/kazu/steps/ner/llm_ner.py
index e8cccbf3..0bbef123 100644
--- a/kazu/steps/ner/llm_ner.py
+++ b/kazu/steps/ner/llm_ner.py
@@ -72,7 +72,9 @@ def parse_result(result: str) -> Optional[dict[str, str]]:
 class AzureOpenAILLMModel(LLMModel):
     """A class to interact with the Azure OpenAI API for LLMs."""
 
-    def __init__(self, model: str, deployment: str, api_version: str, sys_prompt: str, temp: float):
+    def __init__(
+        self, model: str, deployment: str, api_version: str, sys_prompt: str, temp: float
+    ) -> None:
         """Initialize the AzureOpenAILLMModel.
 
         :param model: The model to use.
@@ -117,7 +119,7 @@ def __init__(
         generation_config: dict[str, Any],
         location: str,
         safety_settings: Optional[SafetySettingsType] = None,
-    ):
+    ) -> None:
         """Initialize the VertexLLMModel.
 
         :param project: The project to use.