I can't decide what would be better--have him run off later to join the witch, or earn his redemption by getting punished for his shitty attitude directly. Like, instead of him doodling on the lion statue in the Witch's Garden, he invokes Persephone's name like a big shot but instead he nearly gets mobbed by angry ghosts and it puts the fear of God(desses) into him. Maybe he'll get called Edmund the Wise because the first lesson he learned was how to keep his mouth shut.