fix to addition of WZ stuff