diff --git a/src/compute_horde_prompt_gen/download_model.py b/src/compute_horde_prompt_gen/download_model.py
index 15563cf..c354270 100644
--- a/src/compute_horde_prompt_gen/download_model.py
+++ b/src/compute_horde_prompt_gen/download_model.py
@@ -42,5 +42,5 @@
     )
     model.save_pretrained(args.model_path)
 
-    tokenizer = AutoTokenizer.from_pretrained(args.model_name)
+    tokenizer = AutoTokenizer.from_pretrained(MODEL_PATHS[args.model_name])
     tokenizer.save_pretrained(args.model_path)