diff --git a/packages/core/src/shaders/picking.frag.glsl b/packages/core/src/shaders/picking.frag.glsl
index 5b1864759e..0e58d7eb15 100644
--- a/packages/core/src/shaders/picking.frag.glsl
+++ b/packages/core/src/shaders/picking.frag.glsl
@@ -16,11 +16,11 @@ vec4 filterHighlightColor(vec4 color) {
   if (selected) {
     vec4 highLightColor = u_HighlightColor * COLOR_SCALE;
 
-    // float highLightAlpha = highLightColor.a;
-    // float highLightRatio = highLightAlpha / (highLightAlpha + color.a * (1.0 - highLightAlpha));
+    float highLightAlpha = highLightColor.a;
+    float highLightRatio = highLightAlpha / (highLightAlpha + color.a * (1.0 - highLightAlpha));
 
-    // vec3 resultRGB = mix(color.rgb, highLightColor.rgb, highLightRatio);
-    return highLightColor;
+    vec3 resultRGB = mix(color.rgb, highLightColor.rgb, highLightRatio);
+    return vec4(resultRGB, color.a);
   } else {
     return color;
   }