fix error in InputDataPipeline