diff --git a/process.go b/process.go index ef11a63b0..e51367aaa 100644 --- a/process.go +++ b/process.go @@ -120,6 +120,7 @@ func (p *process) Kill(ctx context.Context, s syscall.Signal, opts ...KillOpts) func (p *process) Wait(ctx context.Context) (<-chan ExitStatus, error) { c := make(chan ExitStatus, 1) go func() { + defer close(c) r, err := p.task.client.TaskService().Wait(ctx, &tasks.WaitRequest{ ContainerID: p.task.id, ExecID: p.id, diff --git a/task.go b/task.go index bf830d2cf..86a00cc48 100644 --- a/task.go +++ b/task.go @@ -215,6 +215,7 @@ func (t *task) Status(ctx context.Context) (Status, error) { func (t *task) Wait(ctx context.Context) (<-chan ExitStatus, error) { c := make(chan ExitStatus, 1) go func() { + defer close(c) r, err := t.client.TaskService().Wait(ctx, &tasks.WaitRequest{ ContainerID: t.id, })