Fixed bug with datasets.