diff --git a/src/freedreno/vulkan/tu_image.cc b/src/freedreno/vulkan/tu_image.cc
index 0278c8f4bf3..672d9fa3712 100644
--- a/src/freedreno/vulkan/tu_image.cc
+++ b/src/freedreno/vulkan/tu_image.cc
@@ -227,8 +227,7 @@ tu_image_view_init(struct tu_device *device,
    }
 
    if (aspect_mask == VK_IMAGE_ASPECT_COLOR_BIT &&
-       (vk_format == VK_FORMAT_G8_B8R8_2PLANE_420_UNORM ||
-        vk_format == VK_FORMAT_G8_B8_R8_3PLANE_420_UNORM)) {
+       vk_format_get_plane_count(vk_format) > 1) {
       layouts[1] = &image->layout[1];
       layouts[2] = &image->layout[2];
    }