diff --git a/websocket/src/handshake.cpp b/websocket/src/handshake.cpp
index 5d5fdcd..a00ee98 100644
--- a/websocket/src/handshake.cpp
+++ b/websocket/src/handshake.cpp
@@ -123,7 +123,7 @@ Result ReceiveHeaders(WebsocketConnection* conn)
     dmSocket::Result sr = WaitForSocket(conn, dmSocket::SELECTOR_KIND_READ, SOCKET_WAIT_TIMEOUT);
     if (dmSocket::RESULT_OK != sr)
     {
-        if (dmSocket::RESULT_WOULDBLOCK)
+        if (dmSocket::RESULT_WOULDBLOCK == sr)
         {
             DebugLog(2, "Waiting for socket to be available for reading");
             return RESULT_WOULDBLOCK;
@@ -291,4 +291,4 @@ Result VerifyHeaders(WebsocketConnection* conn)
 }
 #endif
 
-} // namespace
\ No newline at end of file
+} // namespace