Small bugfix on adding of the masks