Fix a problem with coloring function literal