Fix preprocessing of dataset.