Temp fix for WS accumulation bug.

I'm still not sure why the commented code was not working.