Add test for WAI instruction