little more clean up of the test