diff --git a/src/compiler/nir/nir_control_flow.c b/src/compiler/nir/nir_control_flow.c
index 6edf5e941f4..a427ff9c56a 100644
--- a/src/compiler/nir/nir_control_flow.c
+++ b/src/compiler/nir/nir_control_flow.c
@@ -939,6 +939,7 @@ nir_sort_unstructured_blocks(nir_function_impl *impl)
       assert(blocks[i].index == 0);
       assert(blocks[i].block == NULL);
    }
+   impl->num_blocks = count;
 
    foreach_list_typed_safe(nir_cf_node, node, node, &dead_blocks)
       cleanup_cf_node(node, impl);