diff --git a/zed/src/editor/display_map.rs b/zed/src/editor/display_map.rs
index d99cbfaa384a837d0ab065cdbea5b6c291a503e3..629b27d39585fb699c0caff75cd46e7090955a59 100644
--- a/zed/src/editor/display_map.rs
+++ b/zed/src/editor/display_map.rs
@@ -611,7 +611,8 @@ mod tests {
             [
                 ("out".to_string(), Some("fn.name")),
                 ("…\n".to_string(), None),
-                ("  fn\n \n".to_string(), Some("mod.body"))
+                ("  \nfn ".to_string(), Some("mod.body")),
+                ("i\n".to_string(), Some("fn.name"))
             ]
         );
     }