Refactor getToken code.