shader fix?

assets/shaders/canvas_group_flash.gdshader

@@ -17,6 +17,10 @@ void fragment() {
 	if (selected)
 	{
 		COLOR = vec4(mix(text.rgb,HIGHLIGHT_COLOR.rgb,HIGHLIGHT_COLOR.a),text.a);
+		COLOR = vec4(mix(COLOR.rgb,FLASH_COLOR.rgb,FLASH_COLOR.a*blend),COLOR.a);
 	}
+	else
+	{
 		COLOR = vec4(mix(text.rgb,FLASH_COLOR.rgb,FLASH_COLOR.a*blend),text.a);
 	}
+}

assets/shaders/generic_flash.gdshader

@@ -10,6 +10,7 @@ void fragment() {
 	{
 		COLOR = vec4(mix(COLOR.rgb,HIGHLIGHT_COLOR.rgb,HIGHLIGHT_COLOR.a),COLOR.a);
 	}
+
 	COLOR = vec4(mix(COLOR.rgb,FLASH_COLOR.rgb,FLASH_COLOR.a*blend),COLOR.a);
 
 }