diff --git a/llms/mistral/mistral.py b/llms/mistral/mistral.py
index 105a7988..9b9a602a 100644
--- a/llms/mistral/mistral.py
+++ b/llms/mistral/mistral.py
@@ -254,7 +254,7 @@ if __name__ == "__main__":
         default=0.0,
     )
     parser.add_argument(
-        "--tokens_per_eval",
+        "--tokens-per-eval",
         help="The batch size of tokens to generate.",
         type=int,
         default=10,
diff --git a/llms/mixtral/mixtral.py b/llms/mixtral/mixtral.py
index 6aee4031..b1f14706 100644
--- a/llms/mixtral/mixtral.py
+++ b/llms/mixtral/mixtral.py
@@ -299,7 +299,7 @@ if __name__ == "__main__":
         default="In the beginning the Universe was created.",
     )
     parser.add_argument(
-        "--max_tokens",
+        "--max-tokens",
         "-m",
         type=int,
         default=100,