diff --git a/examples/graphbolt/pyg/hetero/node_classification.py b/examples/graphbolt/pyg/hetero/node_classification.py
index bfad1ab0cf4d..d231c94b7cbc 100644
--- a/examples/graphbolt/pyg/hetero/node_classification.py
+++ b/examples/graphbolt/pyg/hetero/node_classification.py
@@ -64,7 +64,7 @@ def create_dataloader(
     #         "author": ["feat"],
     #         "institution": ["feat"],
     #     }
-    node_feature_keys = {"paper": ["feat"]}
+    node_feature_keys = {"paper": ["feat"], "author": ["feat"]}
     if args.dataset == "ogb-lsc-mag240m":
         node_feature_keys["author"] = ["feat"]
         node_feature_keys["institution"] = ["feat"]