diff --git a/formatter.go b/formatter.go
index 2f42ce2..3916f85 100644
--- a/formatter.go
+++ b/formatter.go
@@ -1,8 +1,9 @@
 package main
 
 import (
-	zordTree "g.arns.lt/zordsdavini/zord-tree"
 	"strings"
+
+	zordTree "g.arns.lt/zordsdavini/zord-tree"
 )
 
 func FormatTreeSummery(tree zordTree.Tree) Tree {
@@ -24,11 +25,7 @@ func readTree(tree zordTree.Tree, ts *Tree, isRoot bool) {
 			if ts.Categories == nil {
 				ts.Categories = make(map[string]int32)
 			}
-			if _, ok := ts.Categories[category]; ok {
-				ts.Categories[category]++
-			} else {
-				ts.Categories[category] = 1
-			}
+			ts.Categories[category] = int32(len(subtree.Files))
 		}
 	}