diff --git a/MindFlow/applications/data_driven/navier_stokes/sno3d/src/utils.py b/MindFlow/applications/data_driven/navier_stokes/sno3d/src/utils.py
index 29de110169d55a715495615ad67cd2583b8bb2da..6867cdd2393482c6a996e8512b9ddf966fc3f8f7 100644
--- a/MindFlow/applications/data_driven/navier_stokes/sno3d/src/utils.py
+++ b/MindFlow/applications/data_driven/navier_stokes/sno3d/src/utils.py
@@ -21,8 +21,8 @@ from scipy.interpolate import interp2d
 import matplotlib
 import matplotlib.pyplot as plt
 import matplotlib.animation as animation
-from mindspore import jit_class, Tensor
-from mindflow.loss import RelativeRMSELoss
+from mindspore import jit_class, ops
+from mindflow.core import RelativeRMSELoss
 from mindflow.utils import print_log
 from mindflow.cell import poly_data
 
@@ -33,8 +33,8 @@ class UnitGaussianNormalizer():
     def __init__(self, x, eps=0.00001):
         super(UnitGaussianNormalizer, self).__init__()
 
-        self.mean = Tensor.mean(x)
-        self.std = Tensor.std(x)
+        self.mean = ops.mean(x)
+        self.std = ops.std(x)
         self.eps = eps
 
     def encode(self, x):