diff --git a/builtin/wait.go b/builtin/wait.go
index 5809e7b..cab0bd5 100644
--- a/builtin/wait.go
+++ b/builtin/wait.go
@@ -90,7 +90,7 @@ func NewWait(name string) *WaitBuilder {
 
 func (b *WaitBuilder) Out(out map[string]any) *WaitBuilder {
 	b.task.Input.Output = orchestrator.Expr[map[string]any]{Expr: out}
-	return b
+	return c
 }
 
 func (b *WaitBuilder) Build() orchestrator.Task {