#3 Fix test to reflect actual data