remove unused import
fix flaky unit test