diffusion module: fix some small compilation problems

no guarantees that it works correctly, though...