diff --git a/client.go b/client.go
index 7885ac9..03996d5 100644
--- a/client.go
+++ b/client.go
@@ -91,8 +91,7 @@ func (p *RTSPPuller) Pull() (err error) {
 		p.Error("Play", zap.Error(err))
 		return err
 	}
-	<-p.Done()
-	return p.Err()
+	return p.Wait()
 }
 
 type RTSPPusher struct {