diff --git a/src/ExtenderManager.php b/src/ExtenderManager.php
index 9792a75..427b62d 100644
--- a/src/ExtenderManager.php
+++ b/src/ExtenderManager.php
@@ -67,6 +67,10 @@ public function addConfigFile($configFile)
      */
     public function addServicesFile($servicesFile)
     {
+        $consoleTags = [
+            'drupal.command',
+            'drupal.generator'
+        ];
         $servicesData = $this->parseData($servicesFile);
         if ($this->isValidServicesData($servicesData)) {
             foreach ($servicesData['services'] as $key => $definition) {
@@ -79,7 +83,7 @@ public function addServicesFile($servicesFile)
                         $bootstrap = null;
                         continue;
                     }
-                    if ($tags['name'] != 'drupal.command') {
+                    if (array_search($tags['name'], $consoleTags) === false) {
                         $bootstrap = null;
                         continue;
                     }