Should you be using a switch here instead.