Fix remove undo