diff --git a/bot.py b/bot.py
index 27c7343..7877ed6 100644
--- a/bot.py
+++ b/bot.py
@@ -22,8 +22,8 @@ async def main():
         await dp.start_polling()
     finally:
         await dp.storage.close()
-        await dp.storage.wait_closed()
-        await bot.session.close()
+        bot_session = await bot.get_session()
+        await bot_session.close()
 
 
 if __name__ == '__main__':