diff --git a/src/patch_denoise/bindings/cli.py b/src/patch_denoise/bindings/cli.py
index 465ed3f..8d98845 100644
--- a/src/patch_denoise/bindings/cli.py
+++ b/src/patch_denoise/bindings/cli.py
@@ -120,7 +120,7 @@ def main():
 
     input_data, affine = load_as_array(args.input_file)
     if args.mask == "auto":
-        mask = compute_mask(input_data, time_axis=-1)
+        mask = compute_mask(input_data)
         affine_mask = None
     else:
         mask, affine_mask = load_as_array(args.mask)
diff --git a/src/patch_denoise/bindings/utils.py b/src/patch_denoise/bindings/utils.py
index 264e131..a94920b 100644
--- a/src/patch_denoise/bindings/utils.py
+++ b/src/patch_denoise/bindings/utils.py
@@ -143,11 +143,11 @@ def compute_mask(array, convex=False):
         Array to compute mask for.
     convex : bool, default False
         If True, the mask is convex for each slice.
+
     Returns
     -------
     numpy.ndarray
         Mask for array.
-
     """
     if time_axis is not None:
         mean = array.mean(axis=-1)