diff --git a/cmd/setup/config_test.go b/cmd/setup/config_test.go
index 8cf241cd6a0..ba92ac8fe76 100644
--- a/cmd/setup/config_test.go
+++ b/cmd/setup/config_test.go
@@ -1,6 +1,7 @@
 package setup
 
 import (
+	"context"
 	"encoding/base64"
 	"fmt"
 	"net/http"
@@ -238,7 +239,7 @@ Actions:
 			v := viper.New()
 			v.SetConfigType("yaml")
 			require.NoError(t, v.ReadConfig(strings.NewReader(tt.args.yaml)))
-			got := MustNewConfig(v)
+			got, _ := MustNewConfig(context.Background(), v)
 			tt.want(t, got)
 		})
 	}
diff --git a/cmd/start/config_test.go b/cmd/start/config_test.go
index 53c95d35ab7..15350c1761a 100644
--- a/cmd/start/config_test.go
+++ b/cmd/start/config_test.go
@@ -1,6 +1,7 @@
 package start
 
 import (
+	"context"
 	"encoding/base64"
 	"fmt"
 	"net"
@@ -275,7 +276,7 @@ Actions:
 			v := viper.New()
 			v.SetConfigType("yaml")
 			require.NoError(t, v.ReadConfig(strings.NewReader(tt.args.yaml)))
-			got := MustNewConfig(v)
+			got, _ := MustNewConfig(context.Background(), v)
 			tt.want(t, got)
 		})
 	}