fix sample.input for diffusion