MAINT: Added mask to existing test.