Update citations at the right time

We were updating before the sentence identifiers were updates - which means that we were always of by one step, which was not only wrong but also confusing for users.