diff --git a/scripts/extension/attributes/morphology_impl.py b/scripts/extension/attributes/morphology_impl.py
index 58e86e4823ab8c8b81705d591e5ec5a2c3c6cc46..da9cf6a1aa3406d5da8d84f2ae7b244201438d74 100644
--- a/scripts/extension/attributes/morphology_impl.py
+++ b/scripts/extension/attributes/morphology_impl.py
@@ -481,7 +481,7 @@ def morphology_impl_nucleus(nucleus_segmentation_path, raw_path, chromatin_path,
 
     # convert to pandas table and add column names
     stats = pd.DataFrame(stats)
-    stats.columns = generate_column_names(raw_path, chromatin_path)
+    stats.columns = generate_column_names(raw_path, chromatin_path, None)
 
     return stats
 
@@ -584,7 +584,7 @@ def morphology_impl_cell(cell_segmentation_path, raw_path,
 
     # convert to pandas table and add column names
     stats = pd.DataFrame(stats)
-    stats.columns = generate_column_names(raw_path, None)
+    stats.columns = generate_column_names(raw_path, None, nucleus_segmentation_path)
 
     return stats