diff --git a/src/aliby/baby_sitter.py b/src/aliby/baby_sitter.py
index 2b56c6fa44756542fe9c8191e4ed17c404370ab7..a7d58df51aa2a4d16e9d0a37b1ee2f246ed358ba 100644
--- a/src/aliby/baby_sitter.py
+++ b/src/aliby/baby_sitter.py
@@ -121,7 +121,8 @@ class BabyRunner(StepABC):
             with_edgemasks=with_edgemasks,
             **kwargs,
         )
-        return format_segmentation(segmentation, tp)
+        res = format_segmentation(segmentation, tp)
+        return res
 
 
 def get_modelset_name_from_params(
diff --git a/src/aliby/pipeline.py b/src/aliby/pipeline.py
index 6eaeaa1c29cafafd94e8e30536bcb73b9feee6f7..fa52f6b963d3ee34e5b0ee9376ad124f7b4a9e30 100644
--- a/src/aliby/pipeline.py
+++ b/src/aliby/pipeline.py
@@ -306,6 +306,10 @@ class Pipeline(ProcessABC):
             print("\n---\n" + step + "\n---")
             pprint(config[step])
         print()
+        try:
+            print(f"Using Baby {baby.__version}")
+        except AttributeError:
+            pass
         # extract from configuration
         expt_id = config["general"]["id"]
         distributed = config["general"]["distributed"]