Beautifying some test code