Fix TestToken unit tests