Update ViewGenerator.php