Modify BaseTokenGenerator