I'm really not keen to keep duplicating blocks of code like this. I don't know how to say this other than "not DRY is BAD". It really is. I am really really convinced of it. But still, I have to keep repreating this in PR... Let's try a different way.

Please don't duplicate blocks of code. Everytime you do a {kitten, pony} dies. Seriously.

How about using a variable that you set based on a class condition?