Fix typo in mask.py