Update test to match new test data