diff --git a/swh/objstorage/replayer/tests/test_replay.py b/swh/objstorage/replayer/tests/test_replay.py
--- a/swh/objstorage/replayer/tests/test_replay.py
+++ b/swh/objstorage/replayer/tests/test_replay.py
@@ -9,7 +9,7 @@
 from hypothesis.strategies import sets
 
 from swh.journal.client import JournalClient
-from swh.journal.writer.kafka import KafkaJournalWriter
+from swh.journal.writer import get_journal_writer
 from swh.model.hypothesis_strategies import sha1
 from swh.model.model import Content
 from swh.objstorage.factory import get_objstorage
@@ -40,7 +40,8 @@
     objstorage1 = get_objstorage(cls="memory")
     objstorage2 = get_objstorage(cls="memory")
 
-    writer = KafkaJournalWriter(
+    writer = get_journal_writer(
+        cls="kafka",
         brokers=[kafka_server],
         client_id="kafka_writer",
         prefix=kafka_prefix,