diff --git a/SessionUtilitiesKit/Database/Models/Job.swift b/SessionUtilitiesKit/Database/Models/Job.swift
index 471df30c1..9af678cee 100644
--- a/SessionUtilitiesKit/Database/Models/Job.swift
+++ b/SessionUtilitiesKit/Database/Models/Job.swift
@@ -294,8 +294,8 @@ public struct Job: Codable, Equatable, Identifiable, FetchableRecord, MutablePer
     
     // MARK: - Custom Database Interaction
     
-    public mutating func didInsert(with rowID: Int64, for column: String?) {
-        self.id = rowID
+    public mutating func didInsert(_ inserted: InsertionSuccess) {
+        self.id = inserted.rowID
     }
 }