change water mark

(cherry picked from commit 944902574be600ded277fcc13937e76e44811cf9)
diff --git a/src/transport/TcpTransport.cpp b/src/transport/TcpTransport.cpp
index 10b7673..d297b1a 100644
--- a/src/transport/TcpTransport.cpp
+++ b/src/transport/TcpTransport.cpp
@@ -256,6 +256,10 @@
 
     if (needed > 0) {
       LOG_DEBUG("too little data received with sum = %d", 4 - needed);
+      /**
+       * reset read water mark to 4
+       */
+      event->setWatermark(EV_READ, 4, 0);
       return;
     }
 
@@ -281,10 +285,6 @@
 
       transport->messageReceived(msg, event->getPeerAddrPort());
     }
-    /**
-     * reset read water mark to 4
-     */
-    event->setWatermark(EV_READ, 4, 0);
   }
 }