That's exactly the tact I use with my kids.

Kid: *some bullshit*

Me: "You want to try that again?"

Kid: "Sorry... *more reasonable version*"

Exactly as you say, after one good talking to, all it takes is a clear reminder "You've screwed up like this before; remember what happened then." and they lesson is recalled and reapplied.