diff --git a/tests/plan/hello-europa/main.cue b/tests/plan/hello-europa/main.cue
index 4459013a..2e4b152f 100644
--- a/tests/plan/hello-europa/main.cue
+++ b/tests/plan/hello-europa/main.cue
@@ -1,11 +1,11 @@
 package main
 
 import (
-	"alpha.dagger.io/europa/dagger/engine"
+	"alpha.dagger.io/europa/dagger"
 	"alpha.dagger.io/os"
 )
 
-engine.#Plan & {
+dagger.#Plan & {
 	actions: {
 		sayHello: os.#Container & {
 			command: "echo Hello Europa! > /out.txt"
diff --git a/tests/plan/proxy/incomplete_service.cue b/tests/plan/proxy/incomplete_service.cue
index 33adbc79..a8432ab9 100644
--- a/tests/plan/proxy/incomplete_service.cue
+++ b/tests/plan/proxy/incomplete_service.cue
@@ -7,8 +7,8 @@ import (
 )
 
 engine.#Plan & {
-	// should fail
-	context: services: dockerSocket: {}
+	// should fail due to incomplete service
+	proxy: dockerSocket: {}
 
 	actions: test: #up: [
 		op.#Load & {
@@ -19,7 +19,7 @@ engine.#Plan & {
 
 		op.#Exec & {
 			always: true
-			mount: "/var/run/docker.sock": stream: context.services.dockerSocket.service
+			mount: "/var/run/docker.sock": stream: proxy.dockerSocket.service
 			args: ["docker", "info"]
 		},
 	]
diff --git a/tests/plan/proxy/incomplete_unix.cue b/tests/plan/proxy/incomplete_unix.cue
index 379dc8bc..475368bc 100644
--- a/tests/plan/proxy/incomplete_unix.cue
+++ b/tests/plan/proxy/incomplete_unix.cue
@@ -7,8 +7,8 @@ import (
 )
 
 engine.#Plan & {
-	// should succeed
-	context: services: dockerSocket: unix: string
+	// should fail because incomplete value
+	proxy: dockerSocket: unix: string
 
 	actions: test: #up: [
 		op.#Load & {
@@ -19,7 +19,7 @@ engine.#Plan & {
 
 		op.#Exec & {
 			always: true
-			mount: "/var/run/docker.sock": stream: context.services.dockerSocket.service
+			mount: "/var/run/docker.sock": stream: proxy.dockerSocket.service
 			args: ["docker", "info"]
 		},
 	]
diff --git a/tests/plan/proxy/invalid_schema.cue b/tests/plan/proxy/invalid_schema.cue
index bf19f9d2..d7f5b174 100644
--- a/tests/plan/proxy/invalid_schema.cue
+++ b/tests/plan/proxy/invalid_schema.cue
@@ -8,7 +8,7 @@ import (
 
 engine.#Plan & {
 	// should fail because of misspelled key
-	context: services: dockerSocket: unx: "/var/run/docker.soc"
+	proxy: dockerSocket: unx: "/var/run/docker.sock"
 
 	actions: test: #up: [
 		op.#Load & {
@@ -19,7 +19,7 @@ engine.#Plan & {
 
 		op.#Exec & {
 			always: true
-			mount: "/var/run/docker.sock": stream: context.services.dockerSocket.service
+			mount: "/var/run/docker.sock": stream: proxy.dockerSocket.service
 			args: ["docker", "info"]
 		},
 	]
diff --git a/tests/plan/proxy/invalid_value.cue b/tests/plan/proxy/invalid_value.cue
index 1dcba88c..6844fe61 100644
--- a/tests/plan/proxy/invalid_value.cue
+++ b/tests/plan/proxy/invalid_value.cue
@@ -8,7 +8,7 @@ import (
 
 engine.#Plan & {
 	// should fail because of misspelled value
-	context: services: dockerSocket: unix: "/var/run/docker.soc"
+	proxy: dockerSocket: unix: "/var/run/docker.soc"
 
 	actions: test: #up: [
 		op.#Load & {
@@ -19,7 +19,7 @@ engine.#Plan & {
 
 		op.#Exec & {
 			always: true
-			mount: "/var/run/docker.sock": stream: context.services.dockerSocket.service
+			mount: "/var/run/docker.sock": stream: proxy.dockerSocket.service
 			args: ["docker", "info"]
 		},
 	]
diff --git a/tests/plan/proxy/unix.cue b/tests/plan/proxy/unix.cue
index 3e1fc723..bc3ae101 100644
--- a/tests/plan/proxy/unix.cue
+++ b/tests/plan/proxy/unix.cue
@@ -8,7 +8,7 @@ import (
 
 engine.#Plan & {
 	// should succeed
-	context: services: dockerSocket: unix: "/var/run/docker.sock"
+	proxy: dockerSocket: unix: "/var/run/docker.sock"
 
 	actions: test: #up: [
 		op.#Load & {
@@ -19,7 +19,7 @@ engine.#Plan & {
 
 		op.#Exec & {
 			always: true
-			mount: "/var/run/docker.sock": stream: context.services.dockerSocket.service
+			mount: "/var/run/docker.sock": stream: proxy.dockerSocket.service
 			args: ["docker", "info"]
 		},
 	]