diff --git a/spacy/syntax/stateclass.pyx b/spacy/syntax/stateclass.pyx
index c206ae992..b2c789d02 100644
--- a/spacy/syntax/stateclass.pyx
+++ b/spacy/syntax/stateclass.pyx
@@ -193,6 +193,7 @@ cdef class StateClass:
         cdef int i
         if child > head:
             self._sent[head].r_kids += 1
+            self._sent[head].r_edge = child
             i = 0
             while self.has_head(head) and i < self.length:
                 self._sent[head].r_edge = child