Fixed the color according to the agitation