better test code