Tokenization: Minor bug fix