updated unit tests to fit for the new states