diff --git a/examples/graphbolt/pyg/hetero/node_classification.py b/examples/graphbolt/pyg/hetero/node_classification.py
index f43a6f0609ba..45a8b1eed730 100644
--- a/examples/graphbolt/pyg/hetero/node_classification.py
+++ b/examples/graphbolt/pyg/hetero/node_classification.py
@@ -346,7 +346,7 @@ def parse_args():
             "igb-het-medium",
             "igb-het-large",
             "igb-het-MLPerf",
-            "igb-het"
+            "igb-het",
         ],
         help="Dataset name. Possible values: ogb-lsc-mag240m, igb-het, and igb-het-[tiny|small|medium|large|MLPerf].",
     )