1 files changed, 6 insertions, 1 deletions
diff --git a/src/client/message.rs b/src/client/message.rs
index 757b673..1a454ab 100644
--- a/src/client/message.rs
+++ b/src/client/message.rs
@@ -102,11 +102,16 @@ pub struct ImageUrl {
 #[cfg(test)]
 mod tests {
     use super::*;
+    use crate::config::InputContext;
 
     #[test]
     fn test_serde() {
         assert_eq!(
-            serde_json::to_string(&Message::new(&Input::from_str("Hello World"))).unwrap(),
+            serde_json::to_string(&Message::new(&Input::from_str(
+                "Hello World",
+                InputContext::default()
+            )))
+            .unwrap(),
             "{\"role\":\"user\",\"content\":\"Hello World\"}"
         );
     }