diff --git a/vta/python/vta/transform.py b/vta/python/vta/transform.py
index b1135c0eb007..1e595c8441b2 100644
--- a/vta/python/vta/transform.py
+++ b/vta/python/vta/transform.py
@@ -988,7 +988,7 @@ def _flatten_loop(src_coeff, dst_coeff, extents):
                         rhs = loop_body.value.args[1]
                     else:
                         raise RuntimeError(
-                            "Function call not recognized %s" % (loop_body.value.name)
+                            "Function call not recognized %s" % (loop_body.value.op.name)
                         )
                 elif isinstance(loop_body.value, tvm.tir.BufferLoad):
                     alu_opcode = env.dev.ALU_OPCODE_SHR