Correcting student‘s work with Teams/OneNote on iPad - annoying change to routine

Since the beginning of this school year, a minor change has made working/correcting student‘s material a bit annoying. After correcting a text in my class notebook on Onenote (having gotten there by clicking on the OneNote - Link in a student‘s assignment section in Teams), I am not sent back to the list of the ‚to be corrected assignments‘ as previously the case when going back to the Teams-app, but all the way back to the main assignments page, which makes it necessary for me to click myself back to the desired list. Apart from being annoying, this actually takes up a considerable amount of time when having to correct an entire class‘ worth of texts. Has anyone encountered a similar change?