diff --git a/workflow/Snakefile b/workflow/Snakefile
index d13c7df6..ee5cb561 100644
--- a/workflow/Snakefile
+++ b/workflow/Snakefile
@@ -27,8 +27,8 @@ from sunbeamlib.reports import *
 client = docker.from_env()
 image_name = f"ctbushman/sunbeam:v{__version__}"
 try:
-    client.images.get(image_name)
-except docker.errors.ImageNotFound:
+    client.images.get_registry_data(image_name)
+except docker.errors.NotFound:
     sys.stderr.write(
         f"WARNING: {image_name} not found on DockerHub, using latest tag instead.\n"
     )