1 files changed, 10 insertions, 2 deletions
diff --git a/vespaclient-core/src/main/java/com/yahoo/clientmetrics/RouteMetricSet.java b/vespaclient-core/src/main/java/com/yahoo/clientmetrics/RouteMetricSet.java
index ebf6246b034..96bcfd4239c 100644
--- a/vespaclient-core/src/main/java/com/yahoo/clientmetrics/RouteMetricSet.java
+++ b/vespaclient-core/src/main/java/com/yahoo/clientmetrics/RouteMetricSet.java
@@ -1,6 +1,8 @@
 // Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
 package com.yahoo.clientmetrics;
 
+import com.yahoo.concurrent.SystemTimer;
+import com.yahoo.concurrent.Timer;
 import com.yahoo.messagebus.Reply;
 
 import java.util.HashMap;
@@ -12,6 +14,7 @@ import java.util.Map;
 public class RouteMetricSet {
 
     private final String route;
+    private final Timer timer;
     private final ProgressCallback callback;
     private final Map<Integer, MessageTypeMetricSet> typeMap = new HashMap<>();
 
@@ -20,18 +23,23 @@ public class RouteMetricSet {
         void done(RouteMetricSet route);
     }
 
-    public RouteMetricSet(String route, ProgressCallback callback) {
+    RouteMetricSet(String route, Timer timer, ProgressCallback callback) {
         this.route = route;
+        this.timer = timer;
         this.callback = callback;
     }
 
+    public RouteMetricSet(String route, ProgressCallback callback) {
+        this(route, SystemTimer.INSTANCE, callback);
+    }
+
     public Map<Integer, MessageTypeMetricSet> getMetrics() { return typeMap; }
 
     public void addReply(Reply r) {
         MessageTypeMetricSet type = typeMap.get(r.getMessage().getType());
         if (type == null) {
             String msgName = r.getMessage().getClass().getSimpleName().replace("Message", "");
-            type = new MessageTypeMetricSet(msgName);
+            type = new MessageTypeMetricSet(msgName, timer);
             typeMap.put(r.getMessage().getType(), type);
         }