diff --git a/src/freedreno/vulkan/tu_cmd_buffer.cc b/src/freedreno/vulkan/tu_cmd_buffer.cc
index 1b74e91a459..e37495c7077 100644
--- a/src/freedreno/vulkan/tu_cmd_buffer.cc
+++ b/src/freedreno/vulkan/tu_cmd_buffer.cc
@@ -3256,10 +3256,10 @@ vk2tu_single_stage(VkPipelineStageFlags2 vk_stage, bool dst)
 }
 
 static enum tu_stage
-vk2tu_src_stage(VkPipelineStageFlags vk_stages)
+vk2tu_src_stage(VkPipelineStageFlags2 vk_stages)
 {
    enum tu_stage stage = TU_STAGE_CP;
-   u_foreach_bit (bit, vk_stages) {
+   u_foreach_bit64 (bit, vk_stages) {
       enum tu_stage new_stage = vk2tu_single_stage(1ull << bit, false);
       stage = MAX2(stage, new_stage);
    }
@@ -3268,10 +3268,10 @@ vk2tu_src_stage(VkPipelineStageFlags vk_stages)
 }
 
 static enum tu_stage
-vk2tu_dst_stage(VkPipelineStageFlags vk_stages)
+vk2tu_dst_stage(VkPipelineStageFlags2 vk_stages)
 {
    enum tu_stage stage = TU_STAGE_PS;
-   u_foreach_bit (bit, vk_stages) {
+   u_foreach_bit64 (bit, vk_stages) {
       enum tu_stage new_stage = vk2tu_single_stage(1ull << bit, true);
       stage = MIN2(stage, new_stage);
    }