Fixed bug in new dataset creation