Update AbstractGenerator.php