diff --git a/src/mesa/main/shaderapi.c b/src/mesa/main/shaderapi.c
index 2b4137ec4bd..6c13c2060cd 100644
--- a/src/mesa/main/shaderapi.c
+++ b/src/mesa/main/shaderapi.c
@@ -550,7 +550,9 @@ detach_shader(struct gl_context *ctx, GLuint program, GLuint shader,
                    shProg->Shaders[j]->Stage == MESA_SHADER_TESS_CTRL ||
                    shProg->Shaders[j]->Stage == MESA_SHADER_TESS_EVAL ||
                    shProg->Shaders[j]->Stage == MESA_SHADER_GEOMETRY ||
-                   shProg->Shaders[j]->Stage == MESA_SHADER_FRAGMENT);
+                   shProg->Shaders[j]->Stage == MESA_SHADER_FRAGMENT ||
+                   shProg->Shaders[j]->Stage == MESA_SHADER_TASK ||
+                   shProg->Shaders[j]->Stage == MESA_SHADER_MESH);
             assert(shProg->Shaders[j]->RefCount > 0);
          }
 #endif