diff --git a/vta/src/runtime.cc b/vta/src/runtime.cc
index 79fc0e293a85d..b657d63a95f5b 100644
--- a/vta/src/runtime.cc
+++ b/vta/src/runtime.cc
@@ -1016,7 +1016,7 @@ class CommandQueue {
           elem_bytes = VTA_ACC_ELEM_BYTES;
           break;
       case VTA_MEM_ID_OUT:
-          elem_bytes = VTA_INP_ELEM_BYTES;
+          elem_bytes = VTA_OUT_ELEM_BYTES;
           break;
       default:
           LOG(FATAL) << "Memory id not recognized:" << memory_id;