clean up import on unit test