Fix NUM_TOKENS usage in thrift_tests