diff --git a/swh/storage/tests/test_api_client.py b/swh/storage/tests/test_api_client.py
--- a/swh/storage/tests/test_api_client.py
+++ b/swh/storage/tests/test_api_client.py
@@ -11,10 +11,11 @@
 
 from swh.core.api.tests.server_testing import ServerTestFixture
 from swh.journal import writer
-from swh.storage.in_memory import Storage as InMemoryStorage
 from swh.storage.api.client import RemoteStorage
 import swh.storage.api.server as server
 from swh.storage.api.server import app
+from swh.storage.in_memory import Storage as InMemoryStorage
+import swh.storage.storage
 from swh.storage.tests.test_storage import \
     CommonTestStorage, CommonPropTestStorage, StorageTestDbFixture
 
@@ -31,7 +32,7 @@
         self.journal_writer = journal_writer
         server.storage = None
         self.get_journal_writer = writer.get_journal_writer
-        writer.get_journal_writer = mock_get_journal_writer
+        swh.storage.storage.get_journal_writer = mock_get_journal_writer
 
         # ServerTestFixture needs to have self.objroot for
         # setUp() method, but this field is defined in