Fix mask slicing for several classes
Merge pull request #56 from astrofrog/fix-mask-slicing

Fix mask slicing for several classes