Fix token checker
