[model_validation] Fixed broken test