Fixing function colorization