Fix bad colorization of several functions