diff --git a/source/tests/pt/test_fitting_net.py b/source/tests/pt/test_fitting_net.py
index 0390043770..e12a397347 100644
--- a/source/tests/pt/test_fitting_net.py
+++ b/source/tests/pt/test_fitting_net.py
@@ -11,15 +11,15 @@
 from deepmd.pt.model.task import (
     EnergyFittingNet,
 )
+from deepmd.pt.utils import (
+    env,
+)
 from deepmd.pt.utils.env import (
     GLOBAL_NP_FLOAT_PRECISION,
 )
 from deepmd.tf.fit.ener import (
     EnerFitting,
 )
-from deepmd.pt.utils import (
-    env,
-)
 
 
 class FakeDescriptor: