diff --git a/lib/cpp/src/transport/TSocket.cc b/lib/cpp/src/transport/TSocket.cc
index 55f512e..2161697 100644
--- a/lib/cpp/src/transport/TSocket.cc
+++ b/lib/cpp/src/transport/TSocket.cc
@@ -1,3 +1,4 @@
+#include <config.h>
 #include <sys/socket.h>
 #include <arpa/inet.h>
 #include <netinet/in.h>
@@ -165,11 +166,11 @@
 
     int flags = 0;
 
-    #if HAVE_BITS_SOCKET_H
+    #if defined(MSG_NOSIGNAL)
     // Note the use of MSG_NOSIGNAL to suppress SIGPIPE errors, instead we
     // check for the EPIPE return condition and close the socket in that case
     flags |= MSG_NOSIGNAL;
-    #endif // HAVE_BITS_SOCKET_H
+    #endif // defined(MSG_NOSIGNAL)
 
     int b = send(socket_, buf + sent, len - sent, flags);
     ++g_socket_syscalls;