diff --git a/samples/nucleus/nucleus.py b/samples/nucleus/nucleus.py
index ccaa890a43c4c3bd3b57fd8e201ae0846cec77d1..315fcac3373d8853d8be58060ba8c7f78aeb962c 100644
--- a/samples/nucleus/nucleus.py
+++ b/samples/nucleus/nucleus.py
@@ -226,7 +226,7 @@ class NucleusDataset(utils.Dataset):
         """
         info = self.image_info[image_id]
         # Get mask directory from image path
-        mask_dir = os.path.join(os.path.dirname(os.path.dirname(path)), "masks")
+        mask_dir = os.path.join(os.path.dirname(os.path.dirname(info['path'])), "masks")
 
         # Read mask files from .png image
         mask = []