added clean up on the test