Cleanup

- Fix single text tokenization