diff --git a/zulip_bots/zulip_bots/tests/test_lib.py b/zulip_bots/zulip_bots/tests/test_lib.py
index 9af194a..253d965 100644
--- a/zulip_bots/zulip_bots/tests/test_lib.py
+++ b/zulip_bots/zulip_bots/tests/test_lib.py
@@ -27,7 +27,7 @@ class FakeClient:
             result='success',
         )
 
-    def get_storage(self):
+    def get_storage(self, keys):
         return dict(
             result='success',
             storage=self.storage,
@@ -78,7 +78,7 @@ class LibTest(TestCase):
         val = state_handler.get('key')
         self.assertEqual(val, [1, 2, 3])
 
-    def test_state_handler(self):
+    def test_state_handler_by_mock(self):
         client = MagicMock()
 
         state_handler = StateHandler(client)