fix states hightlight