diff --git a/src/glsl/nir/spirv_to_nir.c b/src/glsl/nir/spirv_to_nir.c
index 74de2e855d1..c9133f741c6 100644
--- a/src/glsl/nir/spirv_to_nir.c
+++ b/src/glsl/nir/spirv_to_nir.c
@@ -2470,6 +2470,7 @@ vtn_handle_preamble_instruction(struct vtn_builder *b, SpvOp opcode,
          break;
 
       case SpvExecutionModeEarlyFragmentTests:
+         assert(b->shader->stage == MESA_SHADER_FRAGMENT);
          b->shader->info.fs.early_fragment_tests = true;
          break;
 
@@ -2479,15 +2480,19 @@ vtn_handle_preamble_instruction(struct vtn_builder *b, SpvOp opcode,
          break;
 
       case SpvExecutionModeDepthReplacing:
+         assert(b->shader->stage == MESA_SHADER_FRAGMENT);
          b->shader->info.fs.depth_layout = FRAG_DEPTH_LAYOUT_ANY;
          break;
       case SpvExecutionModeDepthGreater:
+         assert(b->shader->stage == MESA_SHADER_FRAGMENT);
          b->shader->info.fs.depth_layout = FRAG_DEPTH_LAYOUT_GREATER;
          break;
       case SpvExecutionModeDepthLess:
+         assert(b->shader->stage == MESA_SHADER_FRAGMENT);
          b->shader->info.fs.depth_layout = FRAG_DEPTH_LAYOUT_LESS;
          break;
       case SpvExecutionModeDepthUnchanged:
+         assert(b->shader->stage == MESA_SHADER_FRAGMENT);
          b->shader->info.fs.depth_layout = FRAG_DEPTH_LAYOUT_UNCHANGED;
          break;