Fix colorizing of states