fix colors not set in display

Small PR just related to previous warning removing, the color bug is now fixed.