Add WU tests