diff --git a/chia/data_layer/data_store.py b/chia/data_layer/data_store.py
index 29c27ca70e7f..9ce0a2e59bdf 100644
--- a/chia/data_layer/data_store.py
+++ b/chia/data_layer/data_store.py
@@ -425,9 +425,9 @@ async def get_kvid(self, blob: bytes, store_id: bytes32) -> Optional[KeyOrValueI
             if row is None:
                 return None
 
-            return KeyId(row[0])
+            return KeyOrValueId(row[0])
 
-    async def get_blob_from_kvid(self, kv_id: Union[KeyId, ValueId], store_id: bytes32) -> Optional[bytes]:
+    async def get_blob_from_kvid(self, kv_id: KeyOrValueId, store_id: bytes32) -> Optional[bytes]:
         async with self.db_wrapper.reader() as reader:
             cursor = await reader.execute(
                 "SELECT blob FROM ids WHERE kv_id = ? AND store_id = ?",