diff --git a/examples/quickstart-pytorch/client.py b/examples/quickstart-pytorch/client.py
index ad57645002f8..1edb42d1ec81 100644
--- a/examples/quickstart-pytorch/client.py
+++ b/examples/quickstart-pytorch/client.py
@@ -99,6 +99,7 @@ def apply_transforms(batch):
 parser.add_argument(
     "--node-id",
     choices=[0, 1, 2],
+    required=True,
     type=int,
     help="Partition of the dataset divided into 3 iid partitions created artificially.",
 )