From 2e207c636cdc59e762c90c0f5a3ee02f8583f4b6 Mon Sep 17 00:00:00 2001
From: Andrey Tikhonov <17@itishka.org>
Date: Tue, 6 Jul 2021 01:10:51 +0300
Subject: [PATCH] fix fault on `reset_state` in memory storage (#619)

---
 aiogram/contrib/fsm_storage/memory.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/aiogram/contrib/fsm_storage/memory.py b/aiogram/contrib/fsm_storage/memory.py
index 8950aa8e..a5686a34 100644
--- a/aiogram/contrib/fsm_storage/memory.py
+++ b/aiogram/contrib/fsm_storage/memory.py
@@ -105,6 +105,7 @@ class MemoryStorage(BaseStorage):
         self.data[chat][user]['bucket'].update(bucket, **kwargs)
 
     def _cleanup(self, chat, user):
+        chat, user = self.resolve_address(chat=chat, user=user)
         if self.data[chat][user] == {'state': None, 'data': {}, 'bucket': {}}:
             del self.data[chat][user]
         if not self.data[chat]: