Cleaning up my test case