clean the code for this test up