1 files changed, 13 insertions, 4 deletions
diff --git a/container-search/src/main/java/com/yahoo/search/grouping/request/AvgFunction.java b/container-search/src/main/java/com/yahoo/search/grouping/request/AvgFunction.java
index 55c4d355589..775f6c48d66 100644
--- a/container-search/src/main/java/com/yahoo/search/grouping/request/AvgFunction.java
+++ b/container-search/src/main/java/com/yahoo/search/grouping/request/AvgFunction.java
@@ -2,12 +2,14 @@
 package com.yahoo.search.grouping.request;
 
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * This class represents a min-function in a {@link GroupingExpression}. It evaluates to a number that equals the
  * average of the results of all arguments.
  *
  * @author Simon Thoresen Hult
+ * @author bratseth
  */
 public class AvgFunction extends FunctionNode {
 
@@ -19,11 +21,18 @@ public class AvgFunction extends FunctionNode {
      * @param argN The optional arguments, must evaluate to a number.
      */
     public AvgFunction(GroupingExpression arg1, GroupingExpression arg2, GroupingExpression... argN) {
-        this(asList(arg1, arg2, argN));
+        this(null, null, asList(arg1, arg2, argN));
     }
 
-    private AvgFunction(List<GroupingExpression> args) {
-        super("avg", args);
+    private AvgFunction(String label, Integer level, List<GroupingExpression> args) {
+        super("avg", label, level, args);
+    }
+
+    @Override
+    public AvgFunction copy() {
+        return new AvgFunction(getLabel(),
+                               getLevelOrNull(),
+                               args().stream().map(arg -> arg.copy()).collect(Collectors.toList()));
     }
 
     /**
@@ -37,6 +46,6 @@ public class AvgFunction extends FunctionNode {
         if (args.size() < 2) {
             throw new IllegalArgumentException("Expected 2 or more arguments, got " + args.size() + ".");
         }
-        return new AvgFunction(args);
+        return new AvgFunction(null, null, args);
     }
 }