diff --git a/router/src/infer.rs b/router/src/infer.rs
index d7b9b52b..a45685cb 100644
--- a/router/src/infer.rs
+++ b/router/src/infer.rs
@@ -800,18 +800,22 @@ mod tests {
                 Message {
                     role: "user".to_string(),
                     content: "Hi!".to_string(),
+                    name: None,
                 },
                 Message {
                     role: "assistant".to_string(),
                     content: "Hello how can I help?".to_string(),
+                    name: None,
                 },
                 Message {
                     role: "user".to_string(),
                     content: "What is Deep Learning?".to_string(),
+                    name: None,
                 },
                 Message {
                     role: "assistant".to_string(),
                     content: "magic!".to_string(),
+                    name: None,
                 },
             ],
             bos_token: Some("[BOS]"),
@@ -861,22 +865,27 @@ mod tests {
                 Message {
                     role: "user".to_string(),
                     content: "Hi!".to_string(),
+                    name: None,
                 },
                 Message {
                     role: "user".to_string(),
                     content: "Hi again!".to_string(),
+                    name: None,
                 },
                 Message {
                     role: "assistant".to_string(),
                     content: "Hello how can I help?".to_string(),
+                    name: None,
                 },
                 Message {
                     role: "user".to_string(),
                     content: "What is Deep Learning?".to_string(),
+                    name: None,
                 },
                 Message {
                     role: "assistant".to_string(),
                     content: "magic!".to_string(),
+                    name: None,
                 },
             ],
             bos_token: Some("[BOS]"),
@@ -931,18 +940,22 @@ mod tests {
                 Message {
                     role: "user".to_string(),
                     content: "Hi!".to_string(),
+                    name: None,
                 },
                 Message {
                     role: "assistant".to_string(),
                     content: "Hello how can I help?".to_string(),
+                    name: None,
                 },
                 Message {
                     role: "user".to_string(),
                     content: "What is Deep Learning?".to_string(),
+                    name: None,
                 },
                 Message {
                     role: "assistant".to_string(),
                     content: "magic!".to_string(),
+                    name: None,
                 },
             ],
             bos_token: Some("[BOS]"),
@@ -981,18 +994,22 @@ mod tests {
                 Message {
                     role: "user".to_string(),
                     content: "Hi!".to_string(),
+                    name: None,
                 },
                 Message {
                     role: "assistant".to_string(),
                     content: "Hello how can I help?".to_string(),
+                    name: None,
                 },
                 Message {
                     role: "user".to_string(),
                     content: "What is Deep Learning?".to_string(),
+                    name: None,
                 },
                 Message {
                     role: "assistant".to_string(),
                     content: "magic!".to_string(),
+                    name: None,
                 },
             ],
             bos_token: Some("[BOS]"),