Maybe address their behaviour but not the problem. With situations like this I often ask leading questions rather than tell them how to behave. If that makes sense. Also ask them how they would feel if you acted this way towards them and if you acted this way towards your partner. And if you do act that way nip that in the bud first