From 7c3c0565b4a0ae26ce80b6c42033d34fa60309ea Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Sat, 14 May 2022 17:29:11 +0200
Subject: [PATCH] Fix wrapped function invocations

---
 pyrogram/session/session.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/pyrogram/session/session.py b/pyrogram/session/session.py
index ff195b96..2e90184a 100644
--- a/pyrogram/session/session.py
+++ b/pyrogram/session/session.py
@@ -351,9 +351,11 @@ class Session:
             pass
 
         if isinstance(query, (raw.functions.InvokeWithoutUpdates, raw.functions.InvokeWithTakeout)):
-            query = query.query
+            inner_query = query.query
+        else:
+            inner_query = query
 
-        query_name = ".".join(query.QUALNAME.split(".")[1:])
+        query_name = ".".join(inner_query.QUALNAME.split(".")[1:])
 
         while True:
             try: