EDT-654 Undo button only working once

https://youtrack.mindtouch.us/issue/EDT-654