diff --git a/swh/vault/tests/test_git_bare_cooker.py b/swh/vault/tests/test_git_bare_cooker.py
index 6510f7d514c3f884cdf4f77ecb48c057c3382498..923f77a81af959da02bfd5fcd09e1caa91d7b1f8 100644
--- a/swh/vault/tests/test_git_bare_cooker.py
+++ b/swh/vault/tests/test_git_bare_cooker.py
@@ -42,16 +42,12 @@ from swh.model.model import (
     TimestampWithTimezone,
 )
 from swh.storage import get_storage
-from swh.storage.postgresql.db import Db as StorageBackend
+from swh.storage.postgresql.storage import Storage
 from swh.vault.cookers.git_bare import GitBareCooker
 from swh.vault.in_memory_backend import InMemoryVaultBackend
 
 storage_postgresql_proc = factories.postgresql_proc(
-    load=[
-        partial(
-            initialize_database_for_module, "storage", StorageBackend.current_version
-        )
-    ],
+    load=[partial(initialize_database_for_module, "storage", Storage.current_version)],
 )
 
 storage_postgresql = factories.postgresql("storage_postgresql_proc")