diff --git a/tests/python/unittest/test_tir_transform_thread_sync.py b/tests/python/unittest/test_tir_transform_thread_sync.py index ffdf4b5916c4c..4e42187b366c0 100644 --- a/tests/python/unittest/test_tir_transform_thread_sync.py +++ b/tests/python/unittest/test_tir_transform_thread_sync.py @@ -55,7 +55,7 @@ def test_thread_storage_sync(): mod = run_passes([A, A2], stmt) f = mod["test_kernel0"] - body_list = tvm.tir.stmt_list(f.body.body.body) + body_list = tvm.tir.stmt_list(f.body.body.body.body) assert body_list[1].value.op.same_as(tvm.ir.Op.get("tir.tvm_storage_sync"))