Remove unused imports in test file