diff --git a/libs/appflowy-ai-client/tests/chat_test/completion_test.rs b/libs/appflowy-ai-client/tests/chat_test/completion_test.rs
index d07bf182..3f216b34 100644
--- a/libs/appflowy-ai-client/tests/chat_test/completion_test.rs
+++ b/libs/appflowy-ai-client/tests/chat_test/completion_test.rs
@@ -27,7 +27,8 @@ async fn completion_image_test() {
     .await
     .unwrap();
   let text = collect_stream_text(stream).await;
-  assert!(text.contains("http://localhost:8000"));
+  println!("{}", text);
+  assert!(text.contains("http://localhost"));
 }
 
 #[tokio::test]