Add the ability to keep the masked value in the model