removing code from the test file