diff --git a/python/tvm/exec/disco_worker.py b/python/tvm/exec/disco_worker.py
index 5b20480decd4..7fe94c6cb0df 100644
--- a/python/tvm/exec/disco_worker.py
+++ b/python/tvm/exec/disco_worker.py
@@ -24,7 +24,7 @@
 import tvm
 from tvm_ffi import get_global_func, register_global_func
 from tvm.runtime import Tensor, ShapeTuple, String
-from tvm.runtime.tensor import tensor
+from tvm.runtime import tensor
 
 
 @register_global_func("tests.disco.add_one", override=True)
diff --git a/python/tvm/testing/runner.py b/python/tvm/testing/runner.py
index f2625b28f972..be50cc8707c5 100644
--- a/python/tvm/testing/runner.py
+++ b/python/tvm/testing/runner.py
@@ -32,7 +32,7 @@
 def _args_to_device(args, device):
     import numpy as np
 
-    from tvm.runtime.tensor import Tensor, empty
+    from tvm.runtime import Tensor, empty
 
     uploaded_args = []
     for arg in args:
@@ -46,7 +46,7 @@ def _args_to_device(args, device):
 
 
 def _args_to_numpy(args):
-    from tvm.runtime.tensor import Tensor
+    from tvm.runtime import Tensor
 
     downloaded_args = []
     for arg in args: