diff --git a/tests/tasks/exec/mount_secret.cue b/tests/tasks/exec/mount_secret.cue
index 74bb04ef..96d38edb 100644
--- a/tests/tasks/exec/mount_secret.cue
+++ b/tests/tasks/exec/mount_secret.cue
@@ -5,7 +5,7 @@ import (
 )
 
 engine.#Plan & {
-	context: secrets: testSecret: envvar: "TESTSECRET"
+	inputs: secrets: testSecret: envvar: "TESTSECRET"
 	actions: {
 		image: engine.#Pull & {
 			source: "alpine:3.15.0@sha256:e7d88de73db3d3fd9b2d63aa7f447a10fd0220b7cbf39803c803f2af9ba256b3"
@@ -15,7 +15,7 @@ engine.#Plan & {
 			input: image.output
 			mounts: secret: {
 				dest:     "/run/secrets/test"
-				contents: context.secrets.testSecret.contents
+				contents: inputs.secrets.testSecret.contents
 			}
 			args: [
 				"sh", "-c",
@@ -30,7 +30,7 @@ engine.#Plan & {
 			input: image.output
 			mounts: secret: {
 				dest:     "/run/secrets/test"
-				contents: context.secrets.testSecret.contents
+				contents: inputs.secrets.testSecret.contents
 				uid:      42
 				gid:      24
 				mask:     0o666
diff --git a/tests/tasks/exec/mount_service.cue b/tests/tasks/exec/mount_service.cue
index 2073b6e9..a3b03460 100644
--- a/tests/tasks/exec/mount_service.cue
+++ b/tests/tasks/exec/mount_service.cue
@@ -5,7 +5,7 @@ import (
 )
 
 engine.#Plan & {
-	context: services: dockerSocket: unix: "/var/run/docker.sock"
+	proxy: dockerSocket: unix: "/var/run/docker.sock"
 
 	actions: {
 		image: engine.#Pull & {
@@ -21,7 +21,7 @@ engine.#Plan & {
 			input: imageWithDocker.output
 			mounts: docker: {
 				dest:     "/var/run/docker.sock"
-				contents: context.services.dockerSocket.service
+				contents: proxy.dockerSocket.service
 			}
 			args: ["docker", "info"]
 		}