Fixed bug with color not staying the same once changed