Fix output returns string, not array