Fixed test and returned a string instead of an array