updating tests to include specs for state class