diff --git a/runtime/bindings/python/vm.cc b/runtime/bindings/python/vm.cc
index d5b7cc6..9a268c1 100644
--- a/runtime/bindings/python/vm.cc
+++ b/runtime/bindings/python/vm.cc
@@ -318,7 +318,7 @@
 
       // Extract dims from the buffer view.
       size_t rank = 0;
-      std::vector<int32_t> dims(6);
+      std::vector<iree_hal_dim_t> dims(6);
       iree_status_t status = iree_hal_buffer_view_shape(
           buffer_view, dims.capacity(), dims.data(), &rank);
       if (iree_status_is_out_of_range(status)) {
@@ -436,7 +436,8 @@
       } else if (iree_hal_buffer_view_isa(variant.ref)) {
         auto hal_bv = iree_hal_buffer_view_deref(variant.ref);
         out += "HalBufferView(";
-        std::vector<int32_t> shape(iree_hal_buffer_view_shape_rank(hal_bv));
+        std::vector<iree_hal_dim_t> shape(
+            iree_hal_buffer_view_shape_rank(hal_bv));
         iree_hal_buffer_view_shape(hal_bv, shape.size(), shape.data(), nullptr);
         for (size_t i = 0; i < shape.size(); ++i) {
           if (i > 0) out += 'x';