Fix tests due to updated data