Make consistent with behavior for rows

when single column is specified as collection, then return dataset, if column is specified
is number or keyword, then return it as vector (old behavior)