fixup: heatHaze

Author: illwieckz

src/engine/renderer/tr_shader.cpp

@@ -2649,13 +2649,6 @@ static bool ParseStage( shaderStage_t *stage, const char **text )
 			}
 			else if ( !Q_stricmp( token, "heathazeMap" ) )
 			{
-				// Do not load heatHaze maps when r_heatHaze is disabled.
-				if ( !r_heatHaze->integer )
-				{
-					stage->active = false;
-					return true;
-				}
-
 				stage->type = stageType_t::ST_HEATHAZEMAP;
 				SetNormalFormat( stage, dxNormalFormat );
 			}
@@ -3284,6 +3277,13 @@ static bool ParseStage( shaderStage_t *stage, const char **text )
 	// compute state bits
 	stage->stateBits = colorMaskBits | depthMaskBits | blendSrcBits | blendDstBits | atestBits | depthFuncBits | polyModeBits;
 
+	// Do not load heatHaze maps when r_heatHaze is disabled.
+	if ( stage->type == stageType_t::ST_HEATHAZEMAP && !r_heatHaze->integer )
+	{
+		stage->active = false;
+		return true;
+	}
+
 	// load image
 	if ( loadMap && !LoadMap( stage, buffer, stage->type ) )
 	{