diff --git a/libs/localenv/target_test.go b/libs/localenv/target_test.go
index 3c64d15462..bf03af40eb 100644
--- a/libs/localenv/target_test.go
+++ b/libs/localenv/target_test.go
@@ -97,7 +97,7 @@ func TestResolveBundleServerless(t *testing.T) {
 	ti, err := ResolveTarget(t.Context(), TargetFlags{}, stubCompute{}, BundleTarget{Selected: true, Serverless: true})
 	require.NoError(t, err)
 	assert.Equal(t, "bundle", ti.Source)
-	assert.Equal(t, "serverless/serverless-"+defaultServerlessVersion, ti.EnvKey)
+	assert.Equal(t, EnvKeyForServerless(defaultServerlessVersion), ti.EnvKey)
 }
 
 // jobStubCompute returns distinct values for the first (sparkVersion) and third
@@ -150,7 +150,7 @@ func TestResolveJobServerlessEmptyVersionFallsBackToDefault(t *testing.T) {
 	c := jobStubCompute{isServerless: true, version: ""}
 	ti, err := ResolveTarget(t.Context(), TargetFlags{Job: "42"}, c, BundleTarget{})
 	require.NoError(t, err)
-	assert.Equal(t, "serverless/serverless-"+defaultServerlessVersion, ti.EnvKey)
+	assert.Equal(t, EnvKeyForServerless(defaultServerlessVersion), ti.EnvKey)
 }
 
 func TestValidateTargetFlagsMutuallyExclusive(t *testing.T) {