diff --git a/lib/client-handshake.c b/lib/client-handshake.c
index 76c297d015a495b8e8349de5d011be03abb9ec48..84b744f00da0fc402887ca0eb8ed11f27da1a8c2 100644
--- a/lib/client-handshake.c
+++ b/lib/client-handshake.c
@@ -241,14 +241,33 @@ libwebsocket_client_connect(struct libwebsocket_context *context,
 	} else
 		wsi->c_origin = NULL;
 
+	wsi->c_callback = NULL;
 	if (protocol) {
+		const char *pc;
+		struct libwebsocket_protocols *pp;
+
 		wsi->c_protocol = malloc(strlen(protocol) + 1);
 		if (wsi->c_protocol == NULL)
 			goto oom3;
+
 		strcpy(wsi->c_protocol, protocol);
+
+		pc = protocol;
+		while (*pc && *pc != ',')
+			pc++;
+		n = pc - protocol;
+		pp = context->protocols;
+		while (pp->name && !wsi->c_callback) {
+			if (!strncmp(protocol, pp->name, n))
+				wsi->c_callback = pp->callback;
+			pp++;
+		}
 	} else
 		wsi->c_protocol = NULL;
 
+	if (!wsi->c_callback)
+		wsi->c_callback = context->protocols[0].callback;
+
 	/* set up appropriate masking */
 
 	wsi->xor_mask = xor_no_mask;