added a check over the sigma when diffusion