I've tried a couple of things to get away from this case statement. I don't really like it but not sure how to fix it.