Fix: missing undo