Fix a bug to add TokenVgram