diff --git a/deepmd/jax/fitting/fitting.py b/deepmd/jax/fitting/fitting.py
index 8a9c4cf321..d62681490c 100644
--- a/deepmd/jax/fitting/fitting.py
+++ b/deepmd/jax/fitting/fitting.py
@@ -6,12 +6,12 @@
 from deepmd.dpmodel.fitting.dipole_fitting import DipoleFitting as DipoleFittingNetDP
 from deepmd.dpmodel.fitting.dos_fitting import DOSFittingNet as DOSFittingNetDP
 from deepmd.dpmodel.fitting.ener_fitting import EnergyFittingNet as EnergyFittingNetDP
-from deepmd.dpmodel.fitting.property_fitting import (
-    PropertyFittingNet as PropertyFittingNetDP,
-)
 from deepmd.dpmodel.fitting.polarizability_fitting import (
     PolarFitting as PolarFittingNetDP,
 )
+from deepmd.dpmodel.fitting.property_fitting import (
+    PropertyFittingNet as PropertyFittingNetDP,
+)
 from deepmd.jax.common import (
     ArrayAPIVariable,
     flax_module,
diff --git a/source/tests/array_api_strict/fitting/fitting.py b/source/tests/array_api_strict/fitting/fitting.py
index 8103e4c120..323a49cfe8 100644
--- a/source/tests/array_api_strict/fitting/fitting.py
+++ b/source/tests/array_api_strict/fitting/fitting.py
@@ -6,12 +6,12 @@
 from deepmd.dpmodel.fitting.dipole_fitting import DipoleFitting as DipoleFittingNetDP
 from deepmd.dpmodel.fitting.dos_fitting import DOSFittingNet as DOSFittingNetDP
 from deepmd.dpmodel.fitting.ener_fitting import EnergyFittingNet as EnergyFittingNetDP
-from deepmd.dpmodel.fitting.property_fitting import (
-    PropertyFittingNet as PropertyFittingNetDP,
-)
 from deepmd.dpmodel.fitting.polarizability_fitting import (
     PolarFitting as PolarFittingNetDP,
 )
+from deepmd.dpmodel.fitting.property_fitting import (
+    PropertyFittingNet as PropertyFittingNetDP,
+)
 
 from ..common import (
     to_array_api_strict_array,