diff --git a/pkg/universe.dagger.io/docker/test/run.cue b/pkg/universe.dagger.io/docker/test/run.cue
index 7354d4bc..477b2e27 100644
--- a/pkg/universe.dagger.io/docker/test/run.cue
+++ b/pkg/universe.dagger.io/docker/test/run.cue
@@ -32,8 +32,8 @@ dagger.#Plan & {
 		}
 
 		// Test: export a file
-		exportFile: {
-			image: _image
+		exportFile: docker.#Run & {
+			input: _image
 			command: {
 				name: "sh"
 				flags: "-c": #"""