Jon should get muzzled because it would make him burst into tears of of desperation to make it up to him. That is all.
It should get to a point where just Jonah sighing in disappointment and asking Jon “am I going to have to muzzle you again?” is enough to have him whining and immediately crumbling <333 Jonah letting him get it out of his system and then sighing all dramatically and reaching for it and Jon tripping over himself to say no no please I can be good please I wanna be a good boy dad please-