fix wae tests