Update token_test.go