Editor cursor did not jump back to the right screen after undo the modification

Here is the step to reproduce:

1. Type some text in the current screen.
2. scroll out of the current screen.
3. Use the short cut Ctrl + Z (or Cmd + Z) to undo the added text.

Expected behaviour:
The cursor should jump back to the screen when it show the line got the modification.

Current behaviour:
The modification is undo but the cursor stays in the current screen.