Fix expectations in test due to changed test data