diff --git a/src/intel/compiler/jay/jay_nir.c b/src/intel/compiler/jay/jay_nir.c
index bbb1d5c1e4c..706e3cbebc6 100644
--- a/src/intel/compiler/jay/jay_nir.c
+++ b/src/intel/compiler/jay/jay_nir.c
@@ -414,7 +414,7 @@ jay_process_nir(const struct intel_device_info *devinfo,
                    &prog_data->vue.vue_map, 0, 0);
    } else if (stage == MESA_SHADER_FRAGMENT) {
       assert(key->fs.mesh_input == INTEL_NEVER && "todo");
-      brw_nir_apply_key(pt, &key->base, 32);
+      brw_nir_apply_key(pt, &key->base, simd_width);
       brw_nir_lower_fs_inputs(nir, devinfo, &key->fs);
       brw_nir_lower_fs_outputs(nir);
       JAY_NIR_SNAPSHOT("after_lower_io");