Update MockModel.php