Fix broken tests (test data incorrect)