Add the mask in 2D as well.