diff --git a/src/fasttext_wrapper.py b/src/fasttext_wrapper.py
index 69ba0c9..cdee8ae 100644
--- a/src/fasttext_wrapper.py
+++ b/src/fasttext_wrapper.py
@@ -41,7 +41,7 @@ def load_context(self, context: mlflow.pyfunc.PythonModelContext) -> None:
     def predict(
         self,
         context: mlflow.pyfunc.PythonModelContext,
-        model_input: List,
+        model_input: List[str],
         params: Optional[Dict[str, Any]] = None
     ) -> Tuple:
         """