refactor(unit): Replace switches with if/else
refactor(unit): thanks, brandon