where did undo go in the notes app?

prior to iOS 18 I always used to use Notes to draw simple pictures and whenever I would make a mistake there was an arrow at the top of the page that you could press and undo your action.

Now that I've upgraded to iOS 18 I just tried to draw a picture and the undo button is gone. The only place that it exists is when you are typing a note but not while you are drawing?