Cleanup for the test class