Fixed bug where states wasn't being displayed properly.