Fixed bug with color display