diff --git a/SwissKnife/segmentation.py b/SwissKnife/segmentation.py
index ed3b185..56115db 100644
--- a/SwissKnife/segmentation.py
+++ b/SwissKnife/segmentation.py
@@ -946,6 +946,7 @@ def main():
     type=str,
     default=None,
     help="number of the gpu to use (can be used to run multiple training processes on the same machine)",
+)
 parser.add_argument(
     "--fraction",
     action="store",