1 files changed, 2 insertions, 1 deletions
diff --git a/exec/java-exec/src/main/java/org/apache/drill/exec/planner/fragment/StatsCollector.java b/exec/java-exec/src/main/java/org/apache/drill/exec/planner/fragment/StatsCollector.java
index 680e6a566..49af366ce 100644
--- a/exec/java-exec/src/main/java/org/apache/drill/exec/planner/fragment/StatsCollector.java
+++ b/exec/java-exec/src/main/java/org/apache/drill/exec/planner/fragment/StatsCollector.java
@@ -71,6 +71,7 @@ public class StatsCollector extends AbstractOpWrapperVisitor<Void, RuntimeExcept
 
     for(ExchangeFragmentPair pair : receivingExchangePairs) {
       if (pair.getExchange() == exchange) {
+        //This is the child fragment which is sending data to this fragment.
         Wrapper sendingFragment = planningSet.get(pair.getNode());
         if (sendingFragment.isEndpointsAssignmentDone()) {
           sendingEndpoints.addAll(sendingFragment.getAssignedEndpoints());
@@ -105,7 +106,7 @@ public class StatsCollector extends AbstractOpWrapperVisitor<Void, RuntimeExcept
       stats.addEndpointAffinities(hasAffinity.getOperatorAffinity());
       stats.setDistributionAffinity(hasAffinity.getDistributionAffinity());
     }
-    stats.addCost(op.getCost());
+    stats.addCost(op.getCost().getOutputRowCount());
     for (PhysicalOperator child : op) {
       child.accept(this, wrapper);
     }