Fixed a small bug with color mode