1 files changed, 3 insertions, 2 deletions
diff --git a/vespa-http-client/src/main/java/com/yahoo/vespa/http/client/core/communication/DocumentQueue.java b/vespa-http-client/src/main/java/com/yahoo/vespa/http/client/core/communication/DocumentQueue.java
index f6b3d1fb62a..ce867d001aa 100644
--- a/vespa-http-client/src/main/java/com/yahoo/vespa/http/client/core/communication/DocumentQueue.java
+++ b/vespa-http-client/src/main/java/com/yahoo/vespa/http/client/core/communication/DocumentQueue.java
@@ -3,6 +3,7 @@ package com.yahoo.vespa.http.client.core.communication;
 
 import com.yahoo.vespa.http.client.core.Document;
 
+import java.time.Duration;
 import java.util.ArrayDeque;
 import java.util.ArrayList;
 import java.util.Deque;
@@ -106,13 +107,13 @@ class DocumentQueue {
         return previousState;
     }
 
-    Optional<Document> pollDocumentIfTimedoutInQueue(long localQueueTimeOut) {
+    Optional<Document> pollDocumentIfTimedoutInQueue(Duration localQueueTimeOut) {
         synchronized (queue) {
             if (queue.isEmpty()) {
                 return Optional.empty();
             }
             Document document = queue.peek();
-            if (document.timeInQueueMillis() > localQueueTimeOut) {
+            if (document.timeInQueueMillis() > localQueueTimeOut.toMillis()) {
                 return Optional.of(queue.poll());
             }
             return Optional.empty();