Comment by TZubiri

Wouldn't a slight change in tokenization? (say mapping single digits to single tokens) help with this specific challenge?