Fix tests in token plugification.