clean up a bit of test code