Fixed undo again (probably)