corrected dataset generation