diff --git a/capsule-prototype/js/online-script.js b/capsule-prototype/js/online-script.js
index a22c231673ba126d293699311f674b16078283ec..dd64c9b1619e71190a0dce9b399fad9390e9357f 100644
--- a/capsule-prototype/js/online-script.js
+++ b/capsule-prototype/js/online-script.js
@@ -855,7 +855,7 @@ function openMosaic() {
     container.append(filterdiv);
     // TODO is there a better way to iterate over tags or documents ?
     for ( let [k, v] of  Object.entries(rekall.sortings.colors.categories)) {
-        let categoryName = v.categoryRaw;
+        let categoryName = rekall.sortings.colors.getCategoryName(k)
         let category = $('<div/>').addClass('mosaic_category').css('background', getTagGradientColor(v));
 
         let grid = $('<div/>').addClass('mosaic_category_grid');
diff --git a/capsule-prototype/js/rekall/Sorting.js b/capsule-prototype/js/rekall/Sorting.js
index 720a4893e0c6ec30ce36215da23caff638fea4ae..62550671dccac736d94dc4ba8416ce2627667f91 100644
--- a/capsule-prototype/js/rekall/Sorting.js
+++ b/capsule-prototype/js/rekall/Sorting.js
@@ -335,4 +335,14 @@ Sorting.unmapPosition = function(x) {
 Sorting.prototype.getCategory = function(tag) {
 	var metadata = this.parseMeta(tag.getMetadata(this.metadataKey));
 	return this.categories[metadata];
-}
\ No newline at end of file
+}
+
+Sorting.prototype.getCategoryName = function(metadata) {
+    if ('_rekall/link' === metadata) {
+        return 'Link';
+    }
+    if ('_rekall/marker' === metadata) {
+        return 'Note';
+    }
+    return metadata.substr(7).toUpperCase();
+}