diff --git a/module_build_service/scheduler/batches.py b/module_build_service/scheduler/batches.py
index ef8f4e96..60d1a3fb 100644
--- a/module_build_service/scheduler/batches.py
+++ b/module_build_service/scheduler/batches.py
@@ -67,7 +67,9 @@ def start_build_component(db_session, builder, c):
         c.state_reason = "Failed to build artifact %s: %s" % (c.package, str(e))
         log.exception(e)
         with BUILD_COMPONENT_DB_SESSION_LOCK:
-            c.module_build.transition(conf, models.BUILD_STATES["failed"], failure_type="infra")
+            c.module_build.transition(
+                db_session, conf, models.BUILD_STATES["failed"], failure_type="infra"
+            )
             db_session.commit()
         return
 
@@ -75,7 +77,9 @@ def start_build_component(db_session, builder, c):
         c.state = koji.BUILD_STATES["FAILED"]
         c.state_reason = "Failed to build artifact %s: Builder did not return task ID" % (c.package)
         with BUILD_COMPONENT_DB_SESSION_LOCK:
-            c.module_build.transition(conf, models.BUILD_STATES["failed"], failure_type="infra")
+            c.module_build.transition(
+                db_session, conf, models.BUILD_STATES["failed"], failure_type="infra"
+            )
             db_session.commit()
         return