Still update a connections piece priority even if the peer doesn't have the piece

diff --git a/connection.go b/connection.go
index a140e24c7a50847b7ff4199fc7023ce403b8e657..85f99e79c9158518b0e106d993d88c9d071b642c 100644 (file)
--- a/connection.go
+++ b/connection.go
@@ -585,10 +585,10 @@ func (c *connection) stopRequestingPiece(piece int) {
 }
 
 func (c *connection) updatePiecePriority(piece int) {
+       tpp := c.t.piecePriority(piece)
        if !c.PeerHasPiece(piece) {
-               return
+               tpp = PiecePriorityNone
        }
-       tpp := c.t.piecePriority(piece)
        if tpp == PiecePriorityNone {
                c.stopRequestingPiece(piece)
                return