From 746a4b685fa0f488825c9d882bbf553706a9ae85 Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Sat, 9 Dec 2017 20:34:31 +0100
Subject: [PATCH] Correctly set the update handler when authorizing a new user

---
 pyrogram/client/client.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyrogram/client/client.py b/pyrogram/client/client.py
index 8934345a..855bb1d6 100644
--- a/pyrogram/client/client.py
+++ b/pyrogram/client/client.py
@@ -255,7 +255,6 @@ class Client:
         self.load_session(self.session_name)
 
         self.session = Session(self.dc_id, self.test_mode, self.auth_key, self.config.api_id)
-        self.session.update_handler = self.update_handler
 
         terms = self.session.start()
 
@@ -265,6 +264,7 @@ class Client:
             self.user_id = self.authorize()
             self.save_session()
 
+        self.session.update_handler = self.update_handler
         self.rnd_id = self.session.msg_id
         self.get_dialogs()