fixed a small bug in coloring