diff --git a/libs/template/renderer_test.go b/libs/template/renderer_test.go
index 2c14009ff..6177fc0fa 100644
--- a/libs/template/renderer_test.go
+++ b/libs/template/renderer_test.go
@@ -17,6 +17,7 @@ import (
 	"github.com/databricks/cli/bundle/phases"
 	"github.com/databricks/cli/cmd/root"
 	"github.com/databricks/cli/internal/testutil"
+	"github.com/databricks/cli/libs/dbr"
 	"github.com/databricks/cli/libs/diag"
 	"github.com/databricks/cli/libs/filer"
 	"github.com/databricks/cli/libs/tags"
@@ -40,7 +41,7 @@ func assertFilePermissions(t *testing.T, path string, perm fs.FileMode) {
 }
 
 func assertBuiltinTemplateValid(t *testing.T, template string, settings map[string]any, target string, isServicePrincipal, build bool, tempDir string) {
-	ctx := context.Background()
+	ctx := dbr.MockRuntime(context.Background(), false)
 
 	templateFS, err := fs.Sub(builtinTemplates, path.Join("templates", template))
 	require.NoError(t, err)