diff --git a/retrievalmarket/network/deal_stream.go b/retrievalmarket/network/deal_stream.go
index 549c44eb..f8ef7960 100644
--- a/retrievalmarket/network/deal_stream.go
+++ b/retrievalmarket/network/deal_stream.go
@@ -55,6 +55,6 @@ func (d *DealStream) WriteDealPayment(dpy retrievalmarket.DealPayment) error {
 	return cborutil.WriteCborRPC(d.rw, &dpy)
 }
 
-func (d *DealStream) Close() {
-	d.rw.Close()
+func (d *DealStream) Close() error {
+	return d.rw.Close()
 }
diff --git a/retrievalmarket/network/network.go b/retrievalmarket/network/network.go
index fb303681..62921a54 100644
--- a/retrievalmarket/network/network.go
+++ b/retrievalmarket/network/network.go
@@ -3,7 +3,6 @@ package network
 import (
 	"github.com/filecoin-project/go-fil-components/retrievalmarket"
 	"github.com/libp2p/go-libp2p-core/peer"
-	peer "github.com/libp2p/go-libp2p-peer"
 )
 
 type RetrievalQueryStream interface {
diff --git a/retrievalmarket/network/query_stream.go b/retrievalmarket/network/query_stream.go
index 7b66fca4..145dd2d1 100644
--- a/retrievalmarket/network/query_stream.go
+++ b/retrievalmarket/network/query_stream.go
@@ -46,6 +46,6 @@ func (qs *QueryStream) WriteQueryResponse(qr retrievalmarket.QueryResponse) erro
 	return cborutil.WriteCborRPC(qs.rw, &qr)
 }
 
-func (qs *QueryStream) Close() {
-	qs.rw.Close()
+func (qs *QueryStream) Close() error {
+	return qs.rw.Close()
 }