Making tests slightly nicer