Fix issue with model 'mean_variance'