diff --git a/src/infrastructure/persistence/run_test.go b/src/infrastructure/persistence/run_test.go
index bcde3b3f..ab2b2674 100644
--- a/src/infrastructure/persistence/run_test.go
+++ b/src/infrastructure/persistence/run_test.go
@@ -57,7 +57,7 @@ func TestShouldUpdateRun(t *testing.T) {
 
 	// given
 	mock, tx := mocks.BuildTransaction(context.Background(), t)
-	mock.ExpectExec("UPDATE runs").WithArgs(run.NomadJobID, run.FinishedAt, run.Success, run.Failure).WillReturnResult(pgxmock.NewResult("UPDATE", 1))
+	mock.ExpectExec("UPDATE runs").WithArgs(run.NomadJobID, run.FinishedAt).WillReturnResult(pgxmock.NewResult("UPDATE", 1))
 	mock.ExpectCommit()
 	repository := NewRunRepository(mock)