Fix the array return