diff --git a/src/Command/Generate/AuthenticationProviderCommand.php b/src/Command/Generate/AuthenticationProviderCommand.php
index d4fa96d08..ca1e2cb7c 100644
--- a/src/Command/Generate/AuthenticationProviderCommand.php
+++ b/src/Command/Generate/AuthenticationProviderCommand.php
@@ -99,7 +99,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/BreakPointCommand.php b/src/Command/Generate/BreakPointCommand.php
index d4582c55f..b8b23af98 100644
--- a/src/Command/Generate/BreakPointCommand.php
+++ b/src/Command/Generate/BreakPointCommand.php
@@ -115,7 +115,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/CacheContextCommand.php b/src/Command/Generate/CacheContextCommand.php
index 84ba545d5..8e08ce402 100644
--- a/src/Command/Generate/CacheContextCommand.php
+++ b/src/Command/Generate/CacheContextCommand.php
@@ -119,7 +119,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/CommandCommand.php b/src/Command/Generate/CommandCommand.php
index ff8049a77..b0f9af2e0 100644
--- a/src/Command/Generate/CommandCommand.php
+++ b/src/Command/Generate/CommandCommand.php
@@ -146,10 +146,9 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $interact = $input->getOption('interact');
         $containerAware = $input->getOption('container-aware');
         $services = $input->getOption('services');
-        $yes = $input->hasOption('yes')?$input->getOption('yes'):false;
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io, $yes)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/ControllerCommand.php b/src/Command/Generate/ControllerCommand.php
index 1f671d202..55454b7af 100644
--- a/src/Command/Generate/ControllerCommand.php
+++ b/src/Command/Generate/ControllerCommand.php
@@ -132,10 +132,9 @@ protected function configure()
     protected function execute(InputInterface $input, OutputInterface $output)
     {
         $io = new DrupalStyle($input, $output);
-        $yes = $input->hasOption('yes')?$input->getOption('yes'):false;
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io, $yes)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/EntityBundleCommand.php b/src/Command/Generate/EntityBundleCommand.php
index 4b1a971a7..e7a19c417 100644
--- a/src/Command/Generate/EntityBundleCommand.php
+++ b/src/Command/Generate/EntityBundleCommand.php
@@ -94,7 +94,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/EventSubscriberCommand.php b/src/Command/Generate/EventSubscriberCommand.php
index e5a16e99f..01fbad1c5 100644
--- a/src/Command/Generate/EventSubscriberCommand.php
+++ b/src/Command/Generate/EventSubscriberCommand.php
@@ -137,7 +137,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/FormAlterCommand.php b/src/Command/Generate/FormAlterCommand.php
index bad154914..5dc12ddcb 100644
--- a/src/Command/Generate/FormAlterCommand.php
+++ b/src/Command/Generate/FormAlterCommand.php
@@ -163,7 +163,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/HelpCommand.php b/src/Command/Generate/HelpCommand.php
index 3a1bc7127..0518692d7 100644
--- a/src/Command/Generate/HelpCommand.php
+++ b/src/Command/Generate/HelpCommand.php
@@ -102,7 +102,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/JsTestCommand.php b/src/Command/Generate/JsTestCommand.php
index 1bc6bb125..fb47b67a9 100644
--- a/src/Command/Generate/JsTestCommand.php
+++ b/src/Command/Generate/JsTestCommand.php
@@ -88,10 +88,9 @@ protected function configure()
     protected function execute(InputInterface $input, OutputInterface $output)
     {
         $io = new DrupalStyle($input, $output);
-        $yes = $input->hasOption('yes') ? $input->getOption('yes') : false;
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io, $yes)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/ModuleCommand.php b/src/Command/Generate/ModuleCommand.php
index c2f42ae41..4b41e65a2 100644
--- a/src/Command/Generate/ModuleCommand.php
+++ b/src/Command/Generate/ModuleCommand.php
@@ -172,10 +172,9 @@ protected function configure()
     protected function execute(InputInterface $input, OutputInterface $output)
     {
         $io = new DrupalStyle($input, $output);
-        $yes = $input->hasOption('yes')?$input->getOption('yes'):false;
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io, $yes)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/ModuleFileCommand.php b/src/Command/Generate/ModuleFileCommand.php
index 53579b31a..a401da9c4 100644
--- a/src/Command/Generate/ModuleFileCommand.php
+++ b/src/Command/Generate/ModuleFileCommand.php
@@ -86,7 +86,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io, $yes)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/PluginBlockCommand.php b/src/Command/Generate/PluginBlockCommand.php
index 5fd449064..0ec87b666 100644
--- a/src/Command/Generate/PluginBlockCommand.php
+++ b/src/Command/Generate/PluginBlockCommand.php
@@ -164,7 +164,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/PluginCKEditorButtonCommand.php b/src/Command/Generate/PluginCKEditorButtonCommand.php
index bae908ea5..01f8a0f0d 100644
--- a/src/Command/Generate/PluginCKEditorButtonCommand.php
+++ b/src/Command/Generate/PluginCKEditorButtonCommand.php
@@ -129,7 +129,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/PluginConditionCommand.php b/src/Command/Generate/PluginConditionCommand.php
index 8491f7bac..e61849546 100644
--- a/src/Command/Generate/PluginConditionCommand.php
+++ b/src/Command/Generate/PluginConditionCommand.php
@@ -143,7 +143,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/PluginFieldCommand.php b/src/Command/Generate/PluginFieldCommand.php
index 450a2e690..ba281126b 100644
--- a/src/Command/Generate/PluginFieldCommand.php
+++ b/src/Command/Generate/PluginFieldCommand.php
@@ -167,7 +167,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/PluginFieldFormatterCommand.php b/src/Command/Generate/PluginFieldFormatterCommand.php
index 0a564fa73..cb61bc431 100644
--- a/src/Command/Generate/PluginFieldFormatterCommand.php
+++ b/src/Command/Generate/PluginFieldFormatterCommand.php
@@ -136,7 +136,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/PluginFieldTypeCommand.php b/src/Command/Generate/PluginFieldTypeCommand.php
index bb5791693..b0bae07ad 100644
--- a/src/Command/Generate/PluginFieldTypeCommand.php
+++ b/src/Command/Generate/PluginFieldTypeCommand.php
@@ -140,7 +140,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/PluginFieldWidgetCommand.php b/src/Command/Generate/PluginFieldWidgetCommand.php
index 01a2ccdb2..9d084ac3f 100644
--- a/src/Command/Generate/PluginFieldWidgetCommand.php
+++ b/src/Command/Generate/PluginFieldWidgetCommand.php
@@ -136,7 +136,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/PluginImageEffectCommand.php b/src/Command/Generate/PluginImageEffectCommand.php
index 2c2d12ce6..8d77749b1 100644
--- a/src/Command/Generate/PluginImageEffectCommand.php
+++ b/src/Command/Generate/PluginImageEffectCommand.php
@@ -127,7 +127,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/PluginImageFormatterCommand.php b/src/Command/Generate/PluginImageFormatterCommand.php
index 486f4d176..337dd751b 100644
--- a/src/Command/Generate/PluginImageFormatterCommand.php
+++ b/src/Command/Generate/PluginImageFormatterCommand.php
@@ -116,7 +116,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/PluginMailCommand.php b/src/Command/Generate/PluginMailCommand.php
index 52366febd..878ee3cbd 100644
--- a/src/Command/Generate/PluginMailCommand.php
+++ b/src/Command/Generate/PluginMailCommand.php
@@ -130,7 +130,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/PluginMigrateProcessCommand.php b/src/Command/Generate/PluginMigrateProcessCommand.php
index dd48a49ff..6e30c7214 100644
--- a/src/Command/Generate/PluginMigrateProcessCommand.php
+++ b/src/Command/Generate/PluginMigrateProcessCommand.php
@@ -108,7 +108,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/PluginMigrateSourceCommand.php b/src/Command/Generate/PluginMigrateSourceCommand.php
index 27d255478..29aae7b07 100644
--- a/src/Command/Generate/PluginMigrateSourceCommand.php
+++ b/src/Command/Generate/PluginMigrateSourceCommand.php
@@ -159,7 +159,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/PluginRestResourceCommand.php b/src/Command/Generate/PluginRestResourceCommand.php
index f976e9c84..2da70e7bb 100644
--- a/src/Command/Generate/PluginRestResourceCommand.php
+++ b/src/Command/Generate/PluginRestResourceCommand.php
@@ -137,7 +137,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/PluginRulesActionCommand.php b/src/Command/Generate/PluginRulesActionCommand.php
index 23e481e02..e548e41b2 100644
--- a/src/Command/Generate/PluginRulesActionCommand.php
+++ b/src/Command/Generate/PluginRulesActionCommand.php
@@ -138,7 +138,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/PluginSkeletonCommand.php b/src/Command/Generate/PluginSkeletonCommand.php
index 015655de0..1c1c21072 100644
--- a/src/Command/Generate/PluginSkeletonCommand.php
+++ b/src/Command/Generate/PluginSkeletonCommand.php
@@ -133,7 +133,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $plugins = $this->getPlugins();
 
         // @see use Drupal\Console\Command\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/PluginViewsFieldCommand.php b/src/Command/Generate/PluginViewsFieldCommand.php
index 5020bf12f..2905bf7dc 100644
--- a/src/Command/Generate/PluginViewsFieldCommand.php
+++ b/src/Command/Generate/PluginViewsFieldCommand.php
@@ -129,7 +129,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/PostUpdateCommand.php b/src/Command/Generate/PostUpdateCommand.php
index 8f9851aca..da72c2903 100644
--- a/src/Command/Generate/PostUpdateCommand.php
+++ b/src/Command/Generate/PostUpdateCommand.php
@@ -107,7 +107,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/ProfileCommand.php b/src/Command/Generate/ProfileCommand.php
index 0cda8d2c5..2103ba7a8 100644
--- a/src/Command/Generate/ProfileCommand.php
+++ b/src/Command/Generate/ProfileCommand.php
@@ -141,11 +141,10 @@ protected function execute(InputInterface $input, OutputInterface $output)
     {
         $io = new DrupalStyle($input, $output);
 
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
-
         // Get the profile path and define a profile path if it is null
         // Check that it is an absolute path or otherwise create an absolute path using appRoot
         $profile_path = $input->getOption('profile-path');
diff --git a/src/Command/Generate/RouteSubscriberCommand.php b/src/Command/Generate/RouteSubscriberCommand.php
index ad3eaf72e..a116b1852 100644
--- a/src/Command/Generate/RouteSubscriberCommand.php
+++ b/src/Command/Generate/RouteSubscriberCommand.php
@@ -107,7 +107,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $output = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($output)) {
+        if (!$this->confirmGeneration($output, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/ServiceCommand.php b/src/Command/Generate/ServiceCommand.php
index 20892badb..4b654618d 100644
--- a/src/Command/Generate/ServiceCommand.php
+++ b/src/Command/Generate/ServiceCommand.php
@@ -144,7 +144,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/ThemeCommand.php b/src/Command/Generate/ThemeCommand.php
index b6d011187..075f03201 100644
--- a/src/Command/Generate/ThemeCommand.php
+++ b/src/Command/Generate/ThemeCommand.php
@@ -181,7 +181,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/TwigExtensionCommand.php b/src/Command/Generate/TwigExtensionCommand.php
index 17eab8ec6..19c13e060 100644
--- a/src/Command/Generate/TwigExtensionCommand.php
+++ b/src/Command/Generate/TwigExtensionCommand.php
@@ -132,7 +132,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Generate/UpdateCommand.php b/src/Command/Generate/UpdateCommand.php
index 2fd0bd40f..53ca889ea 100644
--- a/src/Command/Generate/UpdateCommand.php
+++ b/src/Command/Generate/UpdateCommand.php
@@ -106,7 +106,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $io = new DrupalStyle($input, $output);
 
         // @see use Drupal\Console\Command\Shared\ConfirmationTrait::confirmGeneration
-        if (!$this->confirmGeneration($io)) {
+        if (!$this->confirmGeneration($io, $input)) {
             return 1;
         }
 
diff --git a/src/Command/Shared/ConfirmationTrait.php b/src/Command/Shared/ConfirmationTrait.php
index aaf1800e6..e2b11d1b4 100644
--- a/src/Command/Shared/ConfirmationTrait.php
+++ b/src/Command/Shared/ConfirmationTrait.php
@@ -8,6 +8,7 @@
 namespace Drupal\Console\Command\Shared;
 
 use Drupal\Console\Core\Style\DrupalStyle;
+use Symfony\Component\Console\Input\InputInterface;
 
 /**
  * Class ConfirmationTrait
@@ -17,13 +18,16 @@
 trait ConfirmationTrait
 {
     /**
-     * @param DrupalStyle $io
-     * @param bool        $yes
+     * @param  DrupalStyle $io
+     *   Console interface.
+     * @param  InputInterface $input
+     *   Input interface.
      *
      * @return bool
      */
-    public function confirmGeneration(DrupalStyle $io, $yes = false)
+    public function confirmGeneration(DrupalStyle $io, InputInterface $input)
     {
+        $yes = $input->hasOption('yes') ? $input->getOption('yes') : false;
         if ($yes) {
             return $yes;
         }