From 4a8ef9f99c12e6093c3dcaf22e822a035ffb18a8 Mon Sep 17 00:00:00 2001
From: Paul Sokolovsky <pfalcon@users.sourceforge.net>
Date: Sun, 31 Jul 2016 15:59:24 +0300
Subject: [PATCH] umqtt.simple: Be sure to return socket to blocking state
 after check_msg().

---
 umqtt.simple/umqtt/simple.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/umqtt.simple/umqtt/simple.py b/umqtt.simple/umqtt/simple.py
index 65d06a9a..e4d60cae 100644
--- a/umqtt.simple/umqtt/simple.py
+++ b/umqtt.simple/umqtt/simple.py
@@ -112,9 +112,9 @@ class MQTTClient:
     # messages processed internally.
     def wait_msg(self):
         res = self.sock.read(1)
+        self.sock.setblocking(True)
         if res is None:
             return None
-        self.sock.setblocking(True)
         if res == b"":
             raise OSError(-1)
         if res == b"\xd0":  # PINGRESP