1 files changed, 4 insertions, 2 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/PartialAddress.java b/vespajlib/src/main/java/com/yahoo/tensor/PartialAddress.java
index 4eca9c47402..84f26d96725 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/PartialAddress.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/PartialAddress.java
@@ -122,16 +122,18 @@ public class PartialAddress {
             labels = new Object[size];
         }
 
-        public void add(String dimensionName, long label) {
+        public Builder add(String dimensionName, long label) {
             dimensionNames[index] = dimensionName;
             labels[index] = label;
             index++;
+            return this;
         }
 
-        public void add(String dimensionName, String label) {
+        public Builder add(String dimensionName, String label) {
             dimensionNames[index] = dimensionName;
             labels[index] = label;
             index++;
+            return this;
         }
 
         public PartialAddress build() {