Combine these to a single `if/else` too