Study of Momentum Diffusion with the Effect of Adiabatic Focusing