Modified test to reflect goal