diff --git a/src/client.py b/src/client.py
--- a/src/client.py
+++ b/src/client.py
@@ -27,7 +27,7 @@ class Connection(BaseConnection):
 
 		try:
 			self._socket.connect((host,port))
-		except ConnectionRefusedError as e:
+		except (ConnectionRefusedError,OSError) as e:
 			log.exception(e)
 			print("Couldn't connect to {0}:{1}".format(host,port))
 			raise FailedConnection()
diff --git a/src/netnode.py b/src/netnode.py
--- a/src/netnode.py
+++ b/src/netnode.py
@@ -70,3 +70,4 @@ class NetNode:
 		log.info("updating hash tree...")
 		self._tree.batchUpdate(self._newLeaves.items())
 		self._tree.save(self._treeFile)
+		log.info("tree updated")