Okay i know this isnt my place to comment, but Milo currently is in the wrong as they didn't respect your boundaries when it comes to jokes
No matter what it's about, a sign of uncomfort is the immediate sign to stop what you're doing and step back. Neither of you need to figjt or quit over this. Don't antagonize each other. You can always stay friends with communication.