Updated test for wbi