diff --git a/python/chronos/test/bigdl/chronos/model/tf2/test_mtnet_keras.py b/python/chronos/test/bigdl/chronos/model/tf2/test_mtnet_keras.py
index bfeaf22a2982..4eb043739513 100644
--- a/python/chronos/test/bigdl/chronos/model/tf2/test_mtnet_keras.py
+++ b/python/chronos/test/bigdl/chronos/model/tf2/test_mtnet_keras.py
@@ -22,6 +22,7 @@
 from bigdl.chronos.data import TSDataset
 import pandas as pd
 import numpy as np
+import tensorflow as tf
 from numpy.testing import assert_array_almost_equal
 import tensorflow as tf
 
@@ -56,7 +57,6 @@ def get_data(num_samples):
 class TestMTNetKeras(TestCase):
 
     def setup_method(self, method):
-        import tensorflow as tf
         tf.keras.backend.clear_session()
         train_data, test_data = create_data()
         self.x_train, y_train = train_data.to_numpy()