generate_datasets fixes