diff --git a/src/components/prompt-input.tsx b/src/components/prompt-input.tsx
index 2909ed17..61b50317 100644
--- a/src/components/prompt-input.tsx
+++ b/src/components/prompt-input.tsx
@@ -113,6 +113,7 @@ export default function PromptInput(props: PromptInputProps) {
             props.instanceFolder,
           )
           attachment.url = `data:image/png;base64,${base64Data}`
+          attachment.display = display
           addAttachment(props.instanceId, props.sessionId, attachment)
 
           const textarea = textareaRef