clean up imports in the test files