Update the tests broken by the addition of a state