Fix case where only one result returned and result needs to be wrapped as an array