Correct cast error that made half of the diffusion workspace non functional
Merge pull request #170 from ocommowi/diffusion-workspace-bug

Diffusion workspace bug correction