Removed unused imports in tests