1 files changed, 2 insertions, 2 deletions

diff --git a/analysis/scalarmults/epare/standalone/merge_probs.py b/analysis/scalarmults/epare/standalone/merge_probs.py
index 6baf976..d4477ae 100644
--- a/analysis/scalarmults/epare/standalone/merge_probs.py
+++ b/analysis/scalarmults/epare/standalone/merge_probs.py
@@ -59,9 +59,9 @@ def main(compressed):
                         maps[full].merge(prob_map)
                 except EOFError:
                     break
-                except pickle.UnpicklingError:
+                except (pickle.UnpicklingError, zstd.ZstdError):
                     click.echo(
-                        f"Bad unpickling, the probs file {file} is likely truncated."
+                        f"Bad unpickling/decompression, the probs file {file} is likely truncated."
                     )
                     break
             click.echo(f"Loaded {i} probmaps from {file}.")