From 94fc05ddd01c41abfad5e7752a1b5677f41d11a3 Mon Sep 17 00:00:00 2001
From: Shunsuke Shibayama <sbym1346@gmail.com>
Date: Tue, 3 Sep 2024 21:08:27 +0900
Subject: [PATCH] Update error.rs

---
 crates/erg_common/error.rs | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/crates/erg_common/error.rs b/crates/erg_common/error.rs
index 1cf996e2..74205fe7 100644
--- a/crates/erg_common/error.rs
+++ b/crates/erg_common/error.rs
@@ -770,12 +770,12 @@ impl SubMessage {
             Location::Line(lineno) => {
                 let input = e.input();
                 let (_, vbar) = chars.gutters();
-                let code = input
-                    .reread_lines(lineno as usize, lineno as usize)
-                    .remove(0);
+                let codes = input.reread_lines(lineno as usize, lineno as usize);
+                let default = "???".to_string();
+                let code = codes.first().unwrap_or(&default);
                 let mut cxt = StyledStrings::default();
                 cxt.push_str_with_color(&format!(" {lineno} {vbar} "), gutter_color);
-                cxt.push_str(&code);
+                cxt.push_str(code);
                 cxt.push_str("\n");
                 for msg in self.msg.iter() {
                     cxt.push_str(msg);