From 5c9cab926cc194cf4c566a3571f857be85345806 Mon Sep 17 00:00:00 2001
From: Who? <155328415+coddrago@users.noreply.github.com>
Date: Tue, 4 Mar 2025 13:21:36 +0700
Subject: [PATCH] Update security.py

---
 hikka/security.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/hikka/security.py b/hikka/security.py
index b49932b..f752d08 100644
--- a/hikka/security.py
+++ b/hikka/security.py
@@ -36,9 +36,8 @@ from herokutl.hints import EntityLike
 from herokutl.tl.functions.messages import GetFullChatRequest
 from herokutl.tl.types import ChatParticipantAdmin, ChatParticipantCreator, Message
 from herokutl.utils import get_display_name
-import herokutl as tl
 
-from . import main, utils, loader
+from . import main, utils
 from .database import Database
 from .tl_cache import CustomTelegramClient
 from .types import Command
@@ -390,9 +389,10 @@ class SecurityManager:
         if not (config := self.get_flags(func)):
             return False
 
-        if not user_id and isinstance((await self.client.get_entity(message.peer_id)), tl.types.User):
+        if not user_id:
             user_id = message.sender_id
-        if not user_id and isinstance((await self.client.get_entity(message.peer_id)), tl.types.Channel):
+
+        if not user_id:
             user_id = message.peer_id
 
         is_channel = False