diff --git a/src/main/kotlin/no/iktdev/eventi/stores/TaskStore.kt b/src/main/kotlin/no/iktdev/eventi/stores/TaskStore.kt
index 5f9a52a..50d16a6 100644
--- a/src/main/kotlin/no/iktdev/eventi/stores/TaskStore.kt
+++ b/src/main/kotlin/no/iktdev/eventi/stores/TaskStore.kt
@@ -2,6 +2,7 @@ package no.iktdev.eventi.stores
 
 import no.iktdev.eventi.models.Task
 import no.iktdev.eventi.models.store.PersistedTask
+import no.iktdev.eventi.models.store.TaskStatus
 import java.time.Duration
 import java.util.UUID
 
@@ -14,7 +15,7 @@ interface TaskStore {
 
     fun claim(taskId: UUID, workerId: String): Boolean
     fun heartbeat(taskId: UUID)
-    fun markConsumed(taskId: UUID)
+    fun markConsumed(taskId: UUID, status: TaskStatus)
     fun releaseExpiredTasks(timeout: Duration = Duration.ofMinutes(15))
 
     fun getPendingTasks(): List<PersistedTask>
diff --git a/src/test/kotlin/no/iktdev/eventi/InMemoryTaskStore.kt b/src/test/kotlin/no/iktdev/eventi/InMemoryTaskStore.kt
index 8f46048..ec3fafc 100644
--- a/src/test/kotlin/no/iktdev/eventi/InMemoryTaskStore.kt
+++ b/src/test/kotlin/no/iktdev/eventi/InMemoryTaskStore.kt
@@ -39,9 +39,9 @@ open class InMemoryTaskStore : TaskStore {
         update(task.copy(lastCheckIn = LocalDateTime.now()))
     }
 
-    override fun markConsumed(taskId: UUID) {
+    override fun markConsumed(taskId: UUID, status: TaskStatus) {
         val task = findByTaskId(taskId) ?: return
-        update(task.copy(consumed = true, status = TaskStatus.Completed))
+        update(task.copy(consumed = true, status = status))
     }
 
     override fun releaseExpiredTasks(timeout: Duration) {