diff --git a/drivers/block/tegra_virt_storage/tegra_hv_vblk.c b/drivers/block/tegra_virt_storage/tegra_hv_vblk.c
index 19f24dee..9bbc3de9 100644
--- a/drivers/block/tegra_virt_storage/tegra_hv_vblk.c
+++ b/drivers/block/tegra_virt_storage/tegra_hv_vblk.c
@@ -1145,8 +1145,8 @@ static void setup_device(struct vblk_dev *vblkdev)
 		} else {
 			if (vblkdev->config.blk_config.req_ops_supported & VS_BLK_IOCTL_OP_F) {
 				req->mempool_virt = (void *)((uintptr_t)vblkdev->shared_buffer +
-				(uintptr_t)((req_id % max_ioctl_requests) * max_io_bytes));
-				req->mempool_offset = (req_id % max_ioctl_requests) * max_io_bytes;
+				(uintptr_t)((req_id % max_ioctl_requests) * UFS_IOCTL_MAX_SIZE_SUPPORTED));
+				req->mempool_offset = (req_id % max_ioctl_requests) * UFS_IOCTL_MAX_SIZE_SUPPORTED;
 			}
 		}
 		req->mempool_len = max_io_bytes;