Correction Training for Diffusion Models