Updated tests to check for mask lengths