diff --git a/internal/bundle/python_wheel_test.go b/internal/bundle/python_wheel_test.go
index fd5c9acc..bfc2d8b2 100644
--- a/internal/bundle/python_wheel_test.go
+++ b/internal/bundle/python_wheel_test.go
@@ -8,7 +8,7 @@ import (
 	"github.com/stretchr/testify/require"
 )
 
-func runPythonWheelTest(t *testing.T, pythonWheelWrapper bool) {
+func runPythonWheelTest(t *testing.T, sparkVersion string, pythonWheelWrapper bool) {
 	env := internal.GetEnvOrSkipTest(t, "CLOUD_ENV")
 	t.Log(env)
 
@@ -24,7 +24,7 @@ func runPythonWheelTest(t *testing.T, pythonWheelWrapper bool) {
 	bundleRoot, err := initTestTemplate(t, "python_wheel_task", map[string]any{
 		"node_type_id":         nodeTypeId,
 		"unique_id":            uuid.New().String(),
-		"spark_version":        "13.2.x-snapshot-scala2.12",
+		"spark_version":        sparkVersion,
 		"python_wheel_wrapper": pythonWheelWrapper,
 	})
 	require.NoError(t, err)
@@ -44,9 +44,9 @@ func runPythonWheelTest(t *testing.T, pythonWheelWrapper bool) {
 }
 
 func TestAccPythonWheelTaskDeployAndRunWithoutWrapper(t *testing.T) {
-	runPythonWheelTest(t, false)
+	runPythonWheelTest(t, "13.2.x-snapshot-scala2.12", false)
 }
 
 func TestAccPythonWheelTaskDeployAndRunWithWrapper(t *testing.T) {
-	runPythonWheelTest(t, true)
+	runPythonWheelTest(t, "12.2.x-scala2.12", true)
 }