Fix input states colors