diff --git a/src/public/javascripts/services/tree.js b/src/public/javascripts/services/tree.js
index 3acb635b5..2a506fec1 100644
--- a/src/public/javascripts/services/tree.js
+++ b/src/public/javascripts/services/tree.js
@@ -516,7 +516,8 @@ async function createNote(node, parentNoteId, target, isProtected, saveSelection
         refKey: branchEntity.noteId,
         branchId: branchEntity.branchId,
         isProtected: isProtected,
-        extraClasses: await treeBuilder.getExtraClasses(noteEntity)
+        extraClasses: await treeBuilder.getExtraClasses(noteEntity),
+        icon: treeBuilder.getIcon(noteEntity)
     };
 
     if (target === 'after') {
diff --git a/src/public/javascripts/services/tree_builder.js b/src/public/javascripts/services/tree_builder.js
index 60b5257a8..09c564470 100644
--- a/src/public/javascripts/services/tree_builder.js
+++ b/src/public/javascripts/services/tree_builder.js
@@ -159,5 +159,6 @@ async function getExtraClasses(note) {
 export default {
     prepareTree,
     prepareBranch,
-    getExtraClasses
+    getExtraClasses,
+    getIcon
 }
\ No newline at end of file