diff --git a/src/shaders/colorspace.c b/src/shaders/colorspace.c
index 4575e37075fa5e2fb9530f4d7a907adfc0c18118..4b8f67ad862e8f82ee530d4aabce86504a949a11 100644
--- a/src/shaders/colorspace.c
+++ b/src/shaders/colorspace.c
@@ -820,7 +820,8 @@ static void pl_shader_tone_map(struct pl_shader *sh, struct pl_color_space src,
                                struct pl_shader_obj **peak_detect_state,
                                const struct pl_color_map_params *params)
 {
-    GLSL("// pl_shader_tone_map \n");
+    GLSL("// pl_shader_tone_map \n"
+         "{                     \n");
 
     // To prevent discoloration due to out-of-bounds clipping, we need to make
     // sure to reduce the value range as far as necessary to keep the entire
@@ -999,6 +1000,8 @@ static void pl_shader_tone_map(struct pl_shader *sh, struct pl_color_space src,
     // Undo the normalization by `dst_peak`
     if (dst_range > 1.0 && need_norm)
         GLSL("color.rgb *= %f; \n", dst_range);
+
+    GLSL("} \n");
 }
 
 void pl_shader_color_map(struct pl_shader *sh,