correction of mistake in dataset saving