Tidy test code.