diff --git a/ui/src/visualizations/file-tree-evolution/chart/Sunburst.js b/ui/src/visualizations/file-tree-evolution/chart/Sunburst.js
index 82e0267b..8077b0da 100644
--- a/ui/src/visualizations/file-tree-evolution/chart/Sunburst.js
+++ b/ui/src/visualizations/file-tree-evolution/chart/Sunburst.js
@@ -131,7 +131,7 @@ export default class Sunburst extends React.Component {
     return (
       <div>
         <svg ref={svg => this.chartRef = svg}></svg>
-        <div class="fileName">{this.state.selectedFile}</div>
+        <div className="fileName">{this.state.selectedFile}</div>
       </div>
     );
   }
diff --git a/ui/src/visualizations/file-tree-evolution/sagas/index.js b/ui/src/visualizations/file-tree-evolution/sagas/index.js
index a01eb834..618dda41 100644
--- a/ui/src/visualizations/file-tree-evolution/sagas/index.js
+++ b/ui/src/visualizations/file-tree-evolution/sagas/index.js
@@ -78,8 +78,6 @@ function applyCommit(fileTree, commit, commitIndex, contributor) {
 function modifyFile(fileTree, path, file, commitIndex, contributor, prevPath = '', parentPaths = []) {
   if (path.length === 0) {
     fileTree.size = file.lineCount;
-    fileTree.contributor = contributor;
-    fileTree.changeIteration = commitIndex;
   } else {
     const fullPath = !prevPath ? path[0] : prevPath + '/' + path[0]
     if (!fileTree.children) {
@@ -91,12 +89,18 @@ function modifyFile(fileTree, path, file, commitIndex, contributor, prevPath = '
         name: path[0],
         creationIteration: commitIndex,
         fullPath: fullPath,
-        parentPaths: parentPaths
+        parentPaths: parentPaths,
+        contributor: contributor,
+        changeIteration: commitIndex,
       });
       fileTree.childIndexMap[path[0]] = fileTree.children.length - 1;
     }
+    const currentNode = fileTree.children[fileTree.childIndexMap[path[0]]];
+    // Add those two lines to highlight changes on directory level already, instead of file level only.
+    // currentNode.contributor = contributor;
+    // currentNode.changeIteration = commitIndex;
     modifyFile(
-      fileTree.children[fileTree.childIndexMap[path[0]]], 
+      currentNode, 
       path.slice(1), 
       file, 
       commitIndex,