fix to the token test