diff --git a/src/sparcscore/utils/vis.py b/src/sparcscore/utils/vis.py
index 77e4a9a..c9c13a0 100644
--- a/src/sparcscore/utils/vis.py
+++ b/src/sparcscore/utils/vis.py
@@ -3,7 +3,7 @@
 import os
 
 
-def segmentation_mask(
+def plot_segmentation_mask(
     project,
     mask_channel=0,
     image_channel=0,