diff --git a/renderdoc/driver/vulkan/vk_postvs.cpp b/renderdoc/driver/vulkan/vk_postvs.cpp
index 27e1ef7d3..08e10825c 100644
--- a/renderdoc/driver/vulkan/vk_postvs.cpp
+++ b/renderdoc/driver/vulkan/vk_postvs.cpp
@@ -1626,7 +1626,7 @@ void VulkanReplay::FetchVSOut(uint32_t eventId, VulkanRenderState &state)
     // ensure the push range is visible to the compute shader
     for(const VkPushConstantRange &range : oldPush)
     {
-      if(range.stageFlags == VK_SHADER_STAGE_VERTEX_BIT)
+      if(range.stageFlags & VK_SHADER_STAGE_VERTEX_BIT)
       {
         push = range;
         push.stageFlags = VK_SHADER_STAGE_COMPUTE_BIT;