This makes expressions work in the zoom stack as expected, and even to an extent across the plot dialog. It's a bit strange to have undo work across the dialog and in the main view, but after playing with it, I can see why it feels more intuitive.