Fix bug of datasets with only one features