diff --git a/autogl/module/model/dgl/hetero/han.py b/autogl/module/model/dgl/hetero/han.py
index bbf2bf9..b527ede 100644
--- a/autogl/module/model/dgl/hetero/han.py
+++ b/autogl/module/model/dgl/hetero/han.py
@@ -103,6 +103,7 @@ class HAN(nn.Module):
                     "meta_paths",
                     "num_features",
                     "num_class",
+                    "out_key",
                     "num_layers",
                     "hidden",
                     "heads",