diff --git a/examples/dynamic_shape/example_dynamic.py b/examples/dynamic_shape/example_dynamic.py
index a2d5b46ae..cd0f9cb56 100644
--- a/examples/dynamic_shape/example_dynamic.py
+++ b/examples/dynamic_shape/example_dynamic.py
@@ -1,7 +1,6 @@
 import tilelang
 import tilelang.language as T
 import tilelang.testing
-from tilelang import tvm as tvm
 
 
 @tilelang.jit
@@ -17,9 +16,9 @@ def matmul_dynamic_mnk(
     num_stages,
     threads,
 ):
-    M = tvm.te.var("m")
-    N = tvm.te.var("n")
-    K = tvm.te.var("k")
+    M = T.dynamic("m")
+    N = T.dynamic("n")
+    K = T.dynamic("k")
 
     A_shape = (K, M) if trans_A else (M, K)
     B_shape = (N, K) if trans_B else (K, N)