diff --git a/app/worker/strategies/local.py b/app/worker/strategies/local.py
index 15a8ec2..bedf9b3 100644
--- a/app/worker/strategies/local.py
+++ b/app/worker/strategies/local.py
@@ -72,7 +72,7 @@ class LocalStrategy:
     def run_whisper(
         self,
         filepath: str,
-        task: str,
+        task: Literal["translate", "transcribe"],
         config: Optional[schemas.JobConfig],
         job_id: UUID,
     ) -> List[Any]:
diff --git a/pyproject.toml b/pyproject.toml
index febdfa0..6ab13e1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -35,7 +35,7 @@ tooling = [
   "python-dotenv ==1.0.0",
   "pytest ==7.4.0",
   # types
-  "mypy ==1.0.1",
+  "mypy ==1.4.1",
   "types-requests ==2.31.0.1"
 ]