diff --git a/python/paddle/incubate/distributed/models/moe/gate/naive_gate.py b/python/paddle/incubate/distributed/models/moe/gate/naive_gate.py
index 491d1f95e10cb39695b604aaf640fc15773b1610..c3c68685445c80266b4c19f8fb5bac46ecf8b980 100644
--- a/python/paddle/incubate/distributed/models/moe/gate/naive_gate.py
+++ b/python/paddle/incubate/distributed/models/moe/gate/naive_gate.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 #
 # The file has been adapted from the file:
-#     https://github.com/laekov/fastmoe/blob/master/fmoe/gates/gshard_gate.py
+#     https://github.com/laekov/fastmoe/blob/master/fmoe/gates/naive_gate.py
 #     Git commit hash: 295a615aacce7e54a37e7935274ba15e901c78e4
 # We retain the following license from the original files:
 #     Copyright 2021, Jiaao He. All rights reserved.