Great call to remove case statement and make a simple conditional.