diff --git a/test-server/test-server.c b/test-server/test-server.c
index e019b2b..70bbf30 100644
--- a/test-server/test-server.c
+++ b/test-server/test-server.c
@@ -378,7 +378,7 @@
 			enum libwebsocket_callback_reasons reason,
 					       void *user, void *in, size_t len)
 {
-	int n;
+	int n, m;
 	unsigned char buf[LWS_SEND_BUFFER_PRE_PADDING + 512 +
 						  LWS_SEND_BUFFER_POST_PADDING];
 	unsigned char *p = &buf[LWS_SEND_BUFFER_PRE_PADDING];
@@ -394,9 +394,8 @@
 
 	case LWS_CALLBACK_SERVER_WRITEABLE:
 		n = sprintf((char *)p, "%d", pss->number++);
-		/* too small for partial */
-		n = libwebsocket_write(wsi, p, n, LWS_WRITE_TEXT);
-		if (n < 0) {
+		m = libwebsocket_write(wsi, p, n, LWS_WRITE_TEXT);
+		if (m < n) {
 			lwsl_err("ERROR %d writing to di socket\n", n);
 			return -1;
 		}
@@ -483,7 +482,7 @@
 				   LWS_SEND_BUFFER_PRE_PADDING,
 				   ringbuffer[pss->ringbuffer_tail].len,
 								LWS_WRITE_TEXT);
-			if (n < 0) {
+			if (n < ringbuffer[pss->ringbuffer_tail].len) {
 				lwsl_err("ERROR %d writing to mirror socket\n", n);
 				return -1;
 			}