Update auth callback code