Now correctly displaying state