Fixing selector in broken tests