Removed unused imports in test