bug fixes in dataset