diff --git a/examples/kda/chunk_o.py b/examples/kda/chunk_o.py
index 65bac6647..3d92d9a4c 100644
--- a/examples/kda/chunk_o.py
+++ b/examples/kda/chunk_o.py
@@ -3,7 +3,7 @@
 from tilelang.autotuner import autotune
 
 from FLA_KDA.fla_chunk_o import chunk_gla_fwd_o_gk
-from test_utils import compare_tensors
+from test_utils_kda import compare_tensors
 
 import torch