diff --git a/examples/python/CuTeDSL/cute/tvm_ffi/ampere_gemm_with_fake_tensor.py b/examples/python/CuTeDSL/cute/tvm_ffi/ampere_gemm_with_fake_tensor.py
index f7624da3..68e8f81a 100644
--- a/examples/python/CuTeDSL/cute/tvm_ffi/ampere_gemm_with_fake_tensor.py
+++ b/examples/python/CuTeDSL/cute/tvm_ffi/ampere_gemm_with_fake_tensor.py
@@ -29,7 +29,6 @@
 import sys
 import os
 import torch
-import time
 
 import cutlass
 import cutlass.cute as cute
@@ -43,7 +42,7 @@ This example shows how to compile CuTe JIT function with fake tensors then run i
 if __name__ == "__main__":
     # Add the current directory to sys.path
     current_dir = os.path.dirname(os.path.abspath(__file__))
-    sys.path.insert(0, os.path.join(current_dir, ".."))
+    sys.path.insert(0, os.path.join(current_dir, "..", ".."))
 from ampere.tensorop_gemm import TensorOpGemm
 
 
diff --git a/examples/python/CuTeDSL/cute/tvm_ffi/requirements.txt b/examples/python/CuTeDSL/cute/tvm_ffi/requirements.txt
new file mode 100644
index 00000000..8a3f9dcd
--- /dev/null
+++ b/examples/python/CuTeDSL/cute/tvm_ffi/requirements.txt
@@ -0,0 +1,2 @@
+apache-tvm-ffi
+torch-c-dlpack-ext
\ No newline at end of file