1 files changed, 2 insertions, 2 deletions
diff --git a/docker/types/daemon.py b/docker/types/daemon.py
index af3e5bc..10e8101 100644
--- a/docker/types/daemon.py
+++ b/docker/types/daemon.py
@@ -8,7 +8,7 @@ except ImportError:
 from ..errors import DockerException
 
 
-class CancellableStream(object):
+class CancellableStream:
     """
     Stream wrapper for real-time events, logs, etc. from the server.
 
@@ -32,7 +32,7 @@ class CancellableStream(object):
             return next(self._stream)
         except urllib3.exceptions.ProtocolError:
             raise StopIteration
-        except socket.error:
+        except OSError:
             raise StopIteration
 
     next = __next__