Mask error fix
Work before Christmas