diff --git a/tests/src/Cli/Command/ExceptionGeneratorCommandTest.php b/tests/src/Cli/Command/ExceptionGeneratorCommandTest.php
index 0532d24..4f1d388 100644
--- a/tests/src/Cli/Command/ExceptionGeneratorCommandTest.php
+++ b/tests/src/Cli/Command/ExceptionGeneratorCommandTest.php
@@ -67,8 +67,8 @@ public function executeNoParents(): void
         $input->getOption('no-parents')->willReturn(true);
         $input->getOption('template-path')->willReturn(vfsStream::url('cwd/template_path'));
 
-        $input->bind(Argument::type(InputDefinition::class));
-        $input->validate();
+        $input->bind(Argument::type(InputDefinition::class))->shouldBeCalled();
+        $input->validate()->shouldBeCalled();
 
         $output->writeln(
             'Using path for templates: "vfs://cwd/template_path"',
@@ -148,8 +148,8 @@ public function executeParents(): void
         $input->getOption('no-parents')->willReturn(false);
         $input->getOption('template-path')->willReturn(vfsStream::url('cwd/template_path'));
 
-        $input->bind(Argument::type(InputDefinition::class));
-        $input->validate();
+        $input->bind(Argument::type(InputDefinition::class))->shouldBeCalled();
+        $input->validate()->shouldBeCalled();
 
         $output->writeln(
             'Using path for templates: "vfs://cwd/template_path"',