Subset the mask when importing data