diff --git a/http_parser.c b/http_parser.c
index 834c5b2..2c8416b 100644
--- a/http_parser.c
+++ b/http_parser.c
@@ -1339,7 +1339,7 @@ size_t http_parser_execute (http_parser *parser,
         }
 
         // Exit, the rest of the connect is in a different protocol.
-        if (parser->flags & F_UPGRADE || parser->method == HTTP_CONNECT) {
+        if (parser->upgrade) {
           CALLBACK2(message_complete);
           return (p - data);
         }