Change in calling the Diffusion model normalization