fix incorrect logic in check for segmentation masks

fix incorrect logic in check for segmentation masks #671