diff --git a/src/virtio/vulkan/vn_command_buffer.c b/src/virtio/vulkan/vn_command_buffer.c
index 6d31a104c98..433079daa53 100644
--- a/src/virtio/vulkan/vn_command_buffer.c
+++ b/src/virtio/vulkan/vn_command_buffer.c
@@ -1488,7 +1488,7 @@ vn_CmdCopyImageToBuffer(VkCommandBuffer commandBuffer,
     */
    if (buf->wsi.mem) {
       assert(img->wsi.is_prime_blit_src);
-      assert(!img->wsi.blit_mem);
+      assert(!img->wsi.blit_mem || img->wsi.blit_mem == buf->wsi.mem);
       img->wsi.blit_mem = buf->wsi.mem;
    }