My nephew is the same age as Milo and he freaks out when he gets an erection too.

I think its just very uncomfy and unusual for them? And they dont quite know what to do.

No advice though.... maybe distraction