diff --git a/go/performance/utils/dolt_builder/cmd/main.go b/go/performance/utils/dolt_builder/cmd/main.go
index d2404693b9..e38eab30f3 100644
--- a/go/performance/utils/dolt_builder/cmd/main.go
+++ b/go/performance/utils/dolt_builder/cmd/main.go
@@ -15,6 +15,7 @@
 package main
 
 import (
+	"context"
 	"fmt"
 	"log"
 	"os"
@@ -36,7 +37,7 @@ func main() {
 		os.Exit(2)
 	}
 
-	err := builder.Run(commitList)
+	err := builder.Run(context.Background(), commitList)
 	if err != nil {
 		log.Fatal(err)
 	}
diff --git a/go/performance/utils/dolt_builder/run.go b/go/performance/utils/dolt_builder/run.go
index 5465fda463..5e91675b6e 100644
--- a/go/performance/utils/dolt_builder/run.go
+++ b/go/performance/utils/dolt_builder/run.go
@@ -29,9 +29,7 @@ import (
 
 const envDoltBin = "DOLT_BIN"
 
-func Run(commitList []string) error {
-	parentCtx := context.Background()
-
+func Run(parentCtx context.Context, commitList []string) error {
 	doltBin, err := getDoltBin()
 	if err != nil {
 		return err