diff --git a/src/compiler/nir/nir_lower_non_uniform_access.c b/src/compiler/nir/nir_lower_non_uniform_access.c
index bdaffcb5508..f1834c08f51 100644
--- a/src/compiler/nir/nir_lower_non_uniform_access.c
+++ b/src/compiler/nir/nir_lower_non_uniform_access.c
@@ -463,7 +463,7 @@ nir_lower_non_uniform_access_impl(nir_function_impl *impl,
             all_equal_first = nir_iand(&b, all_equal_first, equal_first);
       }
 
-      nir_push_if(&b, all_equal_first);
+      nir_push_if(&b, all_equal_first)->control = nir_selection_control_divergent_always_taken;
 
       util_dynarray_foreach(&data.srcs, struct nu_handle_src, src) {
          for (uint32_t i = 0; i < key->handle_count; i++)