#4 Dataset builder correction