diff --git a/drivers/net/virtio_net.c b/drivers/net/virtio_net.c
index 0b457c81c..b11b7ae00 100644
--- a/drivers/net/virtio_net.c
+++ b/drivers/net/virtio_net.c
@@ -1106,7 +1106,7 @@ static int virtnet_receive(struct receive_queue *rq, int budget)
 		}
 	}
 
-	if (rq->vq->num_free > virtqueue_get_vring_size(rq->vq) / 2) {
+	if (rq->vq->num_free > min((unsigned int)budget, virtqueue_get_vring_size(rq->vq)) / 2) {
 		if (!try_fill_recv(vi, rq, GFP_ATOMIC))
 			schedule_delayed_work(&vi->refill, 0);
 	}