diff --git a/src/khoj/interface/web/assets/markdown-it.min.js b/src/khoj/interface/web/assets/markdown-it.min.js
index 8cac3add..7339517f 100644
--- a/src/khoj/interface/web/assets/markdown-it.min.js
+++ b/src/khoj/interface/web/assets/markdown-it.min.js
@@ -3228,8 +3228,15 @@
     // Encode attributes, e.g. `<img src="foo"`
         result += this.renderAttrs(token);
 
-    // Add a class name for all tokens, e.g. `<img class="foo"`
-        result += ' class="md-level"';
+      if (token.tag === "img" && token.attrs) {
+          for (var i = 0; i < token.attrs.length; i++) {
+            if (token.attrs[i][0] === "src") {
+              if (token.attrs[i][1].includes("avatar")) {
+                result += ' class="md-avatar"';
+              }
+            }
+          }
+      }
     // Add a slash for self-closing tags, e.g. `<img src="foo" /`
         if (token.nesting === 0 && options.xhtmlOut) {
       result += " /";
diff --git a/src/khoj/interface/web/index.html b/src/khoj/interface/web/index.html
index a40cda37..3bdb7b12 100644
--- a/src/khoj/interface/web/index.html
+++ b/src/khoj/interface/web/index.html
@@ -380,7 +380,7 @@
             text-decoration: none;
         }
 
-        img.md-level {
+        img.md-avatar {
             width: 20px;
             height: 20px;
             border-radius: 50%;