diff --git a/package-lock.json b/package-lock.json
index 98b731349..aae362fd2 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,6 +1,6 @@
 {
   "name": "CRviz",
-  "version": "0.3.1-alpha2",
+  "version": "0.3.1-alpha3",
   "lockfileVersion": 1,
   "requires": true,
   "dependencies": {
diff --git a/package.json b/package.json
index c9d4e756e..af4a107a4 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "CRviz",
-  "version": "0.3.1-alpha2",
+  "version": "0.3.1-alpha3",
   "private": true,
   "dependencies": {
     "@fortawesome/fontawesome": "^1.1.3",
diff --git a/src/features/visualization/d3-viz/setup-annotations.js b/src/features/visualization/d3-viz/setup-annotations.js
index 61f734bfd..2670df2f7 100644
--- a/src/features/visualization/d3-viz/setup-annotations.js
+++ b/src/features/visualization/d3-viz/setup-annotations.js
@@ -44,13 +44,13 @@ const setupAnnotations = ({packedData, annotationRoot, colorMap}) =>{
       .classed(className("total-container"), true);;
   const newCircles = newTotalContainer
     .append('g')
-      .classed(className("leafNode"), true)
+      .classed(className("node"), true)
     .append('circle')
       .attr('r', offsetIncrement)
 
   annotations
   .select(`g.${className("total-container")}`)
-  .select(`g.${className("leafNode")}`)
+  .select(`g.${className("node")}`)
   .select('circle')
   .merge(newCircles)
     .attr('cx', (d) => ((d.r + 2*offsetIncrement) * Math.cos(baseAngle + totalAngleOffset*getAngleOfLeafNodeDiameter(d.r, offsetIncrement))))
diff --git a/src/features/visualization/d3-viz/to-hierarchy.js b/src/features/visualization/d3-viz/to-hierarchy.js
index e74c24cb3..9b3005438 100644
--- a/src/features/visualization/d3-viz/to-hierarchy.js
+++ b/src/features/visualization/d3-viz/to-hierarchy.js
@@ -88,7 +88,7 @@ const countChanged = (children) => {
     result += (children[c].CRVIZ._changedCount || 0) + (children[c].CRVIZ._isChanged || 0) +
               (
                 !isNil(children[c].values)
-                  ? countAdded(children[c].values) : 0
+                  ? countChanged(children[c].values) : 0
               );
   }
 
@@ -109,7 +109,7 @@ const countRemoved = (children) => {
     result += (children[c].CRVIZ._removedCount || 0) + (children[c].CRVIZ._isRemoved || 0) +
               (
                 !isNil(children[c].values)
-                  ? countAdded(children[c].values) : 0
+                  ? countRemoved(children[c].values) : 0
               );
   }