Removed unused imports from tests