diff --git a/tests/test_server.py b/tests/test_server.py
index 3d3d0499..61b0e5df 100644
--- a/tests/test_server.py
+++ b/tests/test_server.py
@@ -77,5 +77,8 @@ def test_api_conversation_generate(conv: str, client: FlaskClient):
     )
     assert response.status_code == 200
     msgs = response.get_json()
-    assert len(msgs) == 1
+    assert len(msgs) >= 1
+    assert len(msgs) <= 2
     assert msgs[0]["role"] == "assistant"
+    if len(msgs) == 2:
+        assert msgs[1]["role"] == "system"