Add ability to explicitly set token.