Yes !
Simple and just.
No need to make any ridiculous complex rules.
That is it.
Let the players have their vengeance against reds, not game mechanisms.
If a player decides to go red, he needs to accept being attacked anywhere, even inside towns.
So, that would make his life miserable, and he...