Fixing wf tests