fixed the test to reflect the updated data structure