chore(tests): updated tests to new Token class