Add a working test