Do not rely on deprel==ROOT, rather use head index