diff --git a/tests/test_sharding.py b/tests/test_sharding.py
index <HASH>..<HASH> 100644
--- a/tests/test_sharding.py
+++ b/tests/test_sharding.py
@@ -14,7 +14,7 @@ logger = logging.getLogger(__name__)
 
 import tempfile
 
-from lib import set_storage
+from lib import set_storage, cleanup_storage
 from lib.shards import Shard, Shards
 from lib.hosts import Hosts
 from lib.process import PortPool, HOSTNAME
@@ -42,7 +42,8 @@ class ShardsTestCase(unittest.TestCase):
         PortPool().change_range()
 
     def tearDown(self):
-        self.sh.cleanup()
+        # self.sh.cleanup()
+        cleanup_storage()
         if os.path.exists(self.db_path):
             os.remove(self.db_path)
use cleanup_storage function instead of sh.cleanup in tearDown