more cleanup on the test