Correct cell segmentation for figures