Use token_callback for checking tokens.

Fixes error if user is not found.