diff --git a/gradle/libs.versions.toml b/gradle/libs.versions.toml
index 66b15b55..494f8292 100644
--- a/gradle/libs.versions.toml
+++ b/gradle/libs.versions.toml
@@ -12,7 +12,7 @@ jsoup = "1.17.2"
 jtokkit = "1.1.0"
 junit = "5.11.0"
 kotlin = "2.0.0"
-llm-client = "0.8.14"
+llm-client = "0.8.16"
 okio = "3.9.0"
 tree-sitter = "0.22.6a"
 
diff --git a/src/test/kotlin/ee/carlrobert/codegpt/completions/DefaultCompletionRequestHandlerTest.kt b/src/test/kotlin/ee/carlrobert/codegpt/completions/DefaultCompletionRequestHandlerTest.kt
index 8dd8530a..4a9b20ff 100644
--- a/src/test/kotlin/ee/carlrobert/codegpt/completions/DefaultCompletionRequestHandlerTest.kt
+++ b/src/test/kotlin/ee/carlrobert/codegpt/completions/DefaultCompletionRequestHandlerTest.kt
@@ -146,9 +146,18 @@ class DefaultCompletionRequestHandlerTest : IntegrationTest() {
           true
         )
       listOf(
-        jsonMapResponse("message", jsonMap(e("content", "Hel"), e("role", "assistant"))),
-        jsonMapResponse("message", jsonMap(e("content", "lo"), e("role", "assistant"))),
-        jsonMapResponse("message", jsonMap(e("content", "!"), e("role", "assistant")))
+        jsonMapResponse(
+          e("message", jsonMap(e("content", "Hel"), e("role", "assistant"))),
+          e("done", false)),
+        jsonMapResponse(
+            e("message", jsonMap(e("content", "lo"), e("role", "assistant"))),
+            e("done", false)),
+        jsonMapResponse(
+            e("message", jsonMap(e("content", "!"), e("role", "assistant"))),
+            e("done", false)),
+        jsonMapResponse(
+            e("message", jsonMap(e("content", ""), e("role", "assistant"))),
+            e("done", true))
       )
     })