If I understand correctly, this is the same as the default behaviour for Emacs (on top of which Undo Tree is implemented).