diff --git a/tests/test_graphql/test_api_storage.py b/tests/test_graphql/test_api_storage.py
index 2325f8c6..59b46dbf 100644
--- a/tests/test_graphql/test_api_storage.py
+++ b/tests/test_graphql/test_api_storage.py
@@ -17,16 +17,13 @@ class BlockDevicesMockReturnNone:
     returncode = 0
 
 
-class BlockDevicesMock:
+class BlockDevicesMockReturnTrue:
     """Mock BlockDevices"""
 
     def __init__(self, args, **kwargs):
         self.args = args
         self.kwargs = kwargs
 
-    def get_block_device(self, name: str):
-        return 0
-
     def mount(self):
         return True