Fixed a bug with color :)