This looks weird to me, but I'm not sure an `if/else` above would be any better and I can't think of a great place to put a function to shorten it.