Fix expected data in test