Fix Incorrect Token Name Bug

Tokens in the case statements (from the predictions) had the wrong
representation in the output because they weren't looked up in the mapping. Now
they are.