Solved problem at output_dataset