Update MessageGenerator.php