diff --git a/_examples/ssh-docker/docker.go b/_examples/ssh-docker/docker.go
index 95c676a..6c2b347 100644
--- a/_examples/ssh-docker/docker.go
+++ b/_examples/ssh-docker/docker.go
@@ -107,12 +107,13 @@ func dockerRun(cfg *container.Config, sess ssh.Session) (status int64, cleanup f
 			}
 		}()
 	}
-	statusChan, chanErr := docker.ContainerWait(ctx, res.ID, container.WaitConditionNotRunning)
-	if chanErr != nil {
+	resultC, errC := docker.ContainerWait(ctx, res.ID, container.WaitConditionNotRunning)
+	select {
+	case err = <-errC:
 		return
+	case result := <-resultC:
+		status = result.StatusCode
 	}
-	err = <-chanErr
-	s := <-statusChan
-	status = s.StatusCode
+	err = <-outputErr
 	return
 }