Remove unused imports in tests