diff --git a/plugins/chunter-resources/src/components/Message.svelte b/plugins/chunter-resources/src/components/Message.svelte
index 2cf7101877..d029607590 100644
--- a/plugins/chunter-resources/src/components/Message.svelte
+++ b/plugins/chunter-resources/src/components/Message.svelte
@@ -164,7 +164,7 @@
       <AttachmentRefInput
         bind:this={refInput}
         space={message.space}
-        _class={chunter.class.Comment}
+        _class={message._class}
         objectId={message._id}
         content={message.content}
         showSend={false}
diff --git a/plugins/chunter-resources/src/components/Thread.svelte b/plugins/chunter-resources/src/components/Thread.svelte
index 59012fee83..05622b4e9b 100644
--- a/plugins/chunter-resources/src/components/Thread.svelte
+++ b/plugins/chunter-resources/src/components/Thread.svelte
@@ -195,7 +195,7 @@
     <div class="mr-4 ml-4 mb-4 mt-2">
       <AttachmentRefInput
         space={parent.space}
-        _class={chunter.class.Comment}
+        _class={chunter.class.ThreadMessage}
         objectId={commentId}
         on:message={onMessage}
       />
diff --git a/plugins/chunter-resources/src/components/ThreadView.svelte b/plugins/chunter-resources/src/components/ThreadView.svelte
index 8af0d4eedf..bcf9c1b057 100644
--- a/plugins/chunter-resources/src/components/ThreadView.svelte
+++ b/plugins/chunter-resources/src/components/ThreadView.svelte
@@ -198,7 +198,12 @@
   {/if}
 </div>
 <div class="ref-input">
-  <AttachmentRefInput space={currentSpace} _class={chunter.class.Comment} objectId={commentId} on:message={onMessage} />
+  <AttachmentRefInput
+    space={currentSpace}
+    _class={chunter.class.ThreadMessage}
+    objectId={commentId}
+    on:message={onMessage}
+  />
 </div>
 
 <style lang="scss">