Tidy tests up