diff --git a/deepmd/pd/utils/dataloader.py b/deepmd/pd/utils/dataloader.py
index c3663ff679..0d1d6af337 100644
--- a/deepmd/pd/utils/dataloader.py
+++ b/deepmd/pd/utils/dataloader.py
@@ -40,7 +40,7 @@
     mix_entropy,
 )
 from deepmd.utils import (
-    dp_random,
+    random as dp_random,
 )
 from deepmd.utils.data import (
     DataRequirementItem,