diff --git a/applications/DeepSpeed-Chat/e2e_rlhf.py b/applications/DeepSpeed-Chat/e2e_rlhf.py
index 51616ad6d..5abae153f 100644
--- a/applications/DeepSpeed-Chat/e2e_rlhf.py
+++ b/applications/DeepSpeed-Chat/e2e_rlhf.py
@@ -65,7 +65,7 @@ def parse_args():
         "--reward-model",
         type=lambda x: x.replace("facebook/opt-", ""),
         default="350m",
-        choices=("350m"),
+        choices=("350m", ),
         help="Which facebook/opt-* model to use for Reward (step 2)",
     )
     parser.add_argument(