Fixed issue with assembly that led to erraneous behaviour for small diffusion constants