Maybe a small bug in test

I was learning about StorageRewrite and found the logic a little confusing in test_tir_transform_storage_rewrite.py:test_inplace_rule2

    A0L = s.cache_read(A0, scope_tb, [A2])
    A1L = s.cache_read(A1, scope_tb, [A2])
    A2L = s.cache_read(A2, scope_tb, [B])

although it doesn’t break the test, I am wondering if

    A0L = s.cache_read(A0, scope_tb, [A2])
    A1L = s.cache_read(A1, scope_tb, [A2])
    A2L = s.cache_write(A2, scope_tb)

is the right code. Thanks a lot!