diff --git a/app/src/benchmark/java/org/signal/benchmark/setup/TestMessages.kt b/app/src/benchmark/java/org/signal/benchmark/setup/TestMessages.kt
index ebb51ac96d..014482c4c9 100644
--- a/app/src/benchmark/java/org/signal/benchmark/setup/TestMessages.kt
+++ b/app/src/benchmark/java/org/signal/benchmark/setup/TestMessages.kt
@@ -59,11 +59,11 @@ object TestMessages {
       null
     )
     if (timestamp != null) {
-      TestDbUtils.setMessageReceived(insert, timestamp)
+      TestDbUtils.setMessageReceived(insert.messageId, timestamp)
     }
-    SignalDatabase.messages.markAsSent(insert, true)
+    SignalDatabase.messages.markAsSent(insert.messageId, true)
 
-    return insert
+    return insert.messageId
   }
   fun insertIncomingTextMessage(other: Recipient, body: String, timestamp: Long? = null) {
     val message = IncomingMessage(