diff --git a/mllam_data_prep/__main__.py b/mllam_data_prep/__main__.py
index 6d18106..d131439 100644
--- a/mllam_data_prep/__main__.py
+++ b/mllam_data_prep/__main__.py
@@ -11,7 +11,7 @@
     import psutil
     from dask.diagnostics import ProgressBar
     from dask.distributed import LocalCluster
-except ImportError or ModuleNotFoundError:  # pragma: no cover
+except (ImportError, ModuleNotFoundError):  # pragma: no cover
     DASK_DISTRIBUTED_AVAILABLE = False
 
 if __name__ == "__main__":  # pragma: no cover