diff --git a/mllam_data_prep/cli.py b/mllam_data_prep/cli.py
index 5bd8e11..907c524 100644
--- a/mllam_data_prep/cli.py
+++ b/mllam_data_prep/cli.py
@@ -3,9 +3,10 @@
 """
 import argparse
 import os
-from loguru import logger
 from pathlib import Path
 
+from loguru import logger
+
 from .create_dataset import create_dataset_zarr
 
 # Attempt to import psutil and dask.distributed modules
diff --git a/tests/test_distributed.py b/tests/test_distributed.py
index d2f7ca7..a7b6a9c 100644
--- a/tests/test_distributed.py
+++ b/tests/test_distributed.py
@@ -20,7 +20,7 @@ def distributed():
         importlib.import_module("dask.distributed")
 
         return True
-    except ModuleNotFoundError, ImportError:
+    except (ModuleNotFoundError, ImportError):
         return False