fixed diffusion run key problem