diff --git a/sukuinote/plugins/log_forwards.py b/sukuinote/plugins/log_forwards.py
index 0255614..8936005 100644
--- a/sukuinote/plugins/log_forwards.py
+++ b/sukuinote/plugins/log_forwards.py
@@ -1,6 +1,7 @@
 import html
 import asyncio
 from pyrogram import Client, filters
+from pyrogram.errors.exceptions.flood_420 import FloodWait
 from .. import config, slave, log_errors, app_user_ids
 
 logged = set()
diff --git a/sukuinote/plugins/log_reports.py b/sukuinote/plugins/log_reports.py
index 28dccf7..2a9713e 100644
--- a/sukuinote/plugins/log_reports.py
+++ b/sukuinote/plugins/log_reports.py
@@ -1,6 +1,7 @@
 import html
 import asyncio
 from pyrogram import Client, filters
+from pyrogram.errors.exceptions.flood_420 import FloodWait
 from .. import config, slave, log_errors
 
 reported = set()