Write better tests