diff --git a/src/config.rs b/src/config.rs
index adbf279..0dbfa32 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -441,15 +441,22 @@ mod test {
                 "completion": {
                     "model": "model1",
                     "parameters": {
-                        "messages": [
+                        "contents": [
                             {
-                                "role": "system",
-                                "content": "Test",
-                            },
+                              "role": "user",
+                              "parts":[{
+                               "text": "Pretend you're a snowman and stay in character for each response."}]
+                              },
                             {
-                                "role": "user",
-                                "content": "Test {CONTEXT} - {CODE}"
-                            }
+                              "role": "model",
+                              "parts":[{
+                               "text": "Hello! It's so cold! Isn't that great?"}]
+                              },
+                            {
+                              "role": "user",
+                              "parts":[{
+                               "text": "What's your favorite season of the year?"}]
+                              }
                         ],
                         "max_new_tokens": 32,
                     }
diff --git a/src/transformer_backends/gemini.rs b/src/transformer_backends/gemini.rs
index d2ac562..be85064 100644
--- a/src/transformer_backends/gemini.rs
+++ b/src/transformer_backends/gemini.rs
@@ -259,7 +259,6 @@ mod test {
         dbg!(response.generated_text);
         Ok(())
     }
-    // gemini_chat_do_generate TODO
     #[tokio::test]
     async fn gemini_chat_do_generate() -> anyhow::Result<()> {
         let configuration: config::Gemini = serde_json::from_value(json!({