diff --git a/download_service.py b/download_service.py
index feb4b90..f6b22f1 100644
--- a/download_service.py
+++ b/download_service.py
@@ -11,7 +11,7 @@
 from telethon.tl.types import DocumentAttributeFilename
 from telethon.tl.types import Document
 from telethon.utils import get_input_media
-from telethon.errors.rpc_error_list import LocationInvalidError
+from telethon.errors.rpc_error_list import LocationInvalidError, SessionPasswordNeededError
 # from telegram_client_x import TelegramClientX
 from telethon.telegram_client import TelegramClient
 from telethon.tl.types import Message
@@ -68,7 +68,12 @@ def write(self, *args):
 if not client.is_user_authorized():
     phone = input('Enter phone: ')
     client.send_code_request(phone)
-    client.sign_in(phone, input('Enter code: '))
+    try:
+        client.sign_in(phone, input('Enter code: '))
+    except SessionPasswordNeededError:
+        client.sign_in(password=input('Enter password:'))
+
+
 def on_download_progress(recv_bytes, total_bytes):
     global last_call_time_receive
     if time.time() - last_call_time_receive < 1: