Updating tests to match new state size
More test updates to match new state size