Fix unit test imports.