Replaced if-statement with a switch in pull request review comment.