Fix up the states output