Use a case statement instead