diff --git a/infer.py b/infer.py
index ce29670..45b6689 100644
--- a/infer.py
+++ b/infer.py
@@ -9,7 +9,7 @@
 def parse_args():
     parser = ArgumentParser()
 
-    parser.add_argument('model-path', type=str,
+    parser.add_argument('model_path', type=str,
                         help='path to checkpoint file containing trained model')
     parser.add_argument('-g', '--gpu', type=int, default=0,
                         help='gpu to use for inference (-1 for cpu)')