Update added token_type as required