diff --git a/crates/editor/src/display_map/inlay_map.rs b/crates/editor/src/display_map/inlay_map.rs
index bf9a466f0d..e64f740052 100644
--- a/crates/editor/src/display_map/inlay_map.rs
+++ b/crates/editor/src/display_map/inlay_map.rs
@@ -153,6 +153,7 @@ pub struct InlayChunks<'a> {
     inlay_chunks: Option<text::Chunks<'a>>,
     output_offset: InlayOffset,
     max_output_offset: InlayOffset,
+    highlight_style: Option<HighlightStyle>,
 }
 
 #[derive(Debug, Clone)]
@@ -208,6 +209,7 @@ impl<'a> Iterator for InlayChunks<'a> {
                 self.output_offset.0 += chunk.len();
                 Chunk {
                     text: chunk,
+                    highlight_style: self.highlight_style,
                     ..Default::default()
                 }
             }
@@ -796,6 +798,7 @@ impl InlaySnapshot {
             suggestion_chunk: None,
             output_offset: range.start,
             max_output_offset: range.end,
+            highlight_style: suggestion_highlight,
         }
     }