Fix for model_coefs mask