Fix country flag check warning