Fixing TODO about custom_red