Fixed typo in utils/dataset.py