diff --git a/docker_squash/image.py b/docker_squash/image.py
index 98e4e90..db744f3 100644
--- a/docker_squash/image.py
+++ b/docker_squash/image.py
@@ -354,7 +354,7 @@ def _save_image(self, image_id, directory):
             try:
                 image = self.docker.get_image(image_id)
 
-                if docker.version_info[0] < 3:
+                if int(docker.__version__.split('.')[0]) < 3:
                     # Docker library prior to 3.0.0 returned the requests
                     # object directly which cold be used to read from
                     self.log.debug(