Handle potential multiple inputs in diffusion