diff --git a/plantcv/annotate/classes.py b/plantcv/annotate/classes.py
index 3b7af9b..144f57d 100644
--- a/plantcv/annotate/classes.py
+++ b/plantcv/annotate/classes.py
@@ -163,7 +163,8 @@ class label, by default "default"
             for (x, y) in self.coords[self.label]:
                 self.ax.plot(x, y, marker='x', c=self.color)
 
-    def _create_pts_mask(self, bin_img):
+
+    def _create_pts_mask(self, bin_img, labelnames):
         """Fitler a binary mask based on annotations.
 
         Parameters
@@ -176,7 +177,6 @@ def _create_pts_mask(self, bin_img):
         pts_mask : numpy.ndarray
             binary mask of annotations
         """
-        labelnames = list(self.count)
         pts_mask = np.zeros(np.shape(bin_img), np.uint8)
         # Create points mask from all annotations
         for names in labelnames:
@@ -216,7 +216,9 @@ def correct_mask(self, bin_img):
         added_obj_labels = []
         analysis_labels = []
         
-        pts_mask = self._create_pts_mask(bin_img)
+        labelnames = list(self.count)
+        
+        pts_mask = self._create_pts_mask(bin_img, labelnames)
         
         final_mask = np.zeros(np.shape(bin_img), np.uint32)
         debug_img = np.zeros(np.shape(bin_img), np.uint8)