Shouldn't we use `_get_mask` instead?