Small mod to get_token