diff --git a/Plugin.php b/Plugin.php
index 9f8b3cf..0550a8f 100755
--- a/Plugin.php
+++ b/Plugin.php
@@ -10,7 +10,7 @@ class Plugin implements PluginEntryPointInterface
     /** @return void */
     public function __invoke(RegistrationInterface $psalm, SimpleXMLElement $config = null)
     {
-    	$psalm->addStubFile(__DIR__ . '/stubs/Assert.php');
+        $psalm->addStubFile(__DIR__ . '/stubs/Assert.php');
         $psalm->addStubFile(__DIR__ . '/stubs/TestCase.php');
         $psalm->addStubFile(__DIR__ . '/stubs/MockBuilder.php');
         $psalm->addStubFile(__DIR__ . '/stubs/InvocationMocker.php');
diff --git a/stubs/MockBuilder.php b/stubs/MockBuilder.php
index 68fd24b..2ffcde3 100644
--- a/stubs/MockBuilder.php
+++ b/stubs/MockBuilder.php
@@ -2,6 +2,7 @@
 namespace PHPUnit\Framework\MockObject;
 
 use PHPUnit\Framework\TestCase;
+use PHPUnit\Framework\MockObject\Builder\InvocationMocker;
 
 /**
  * @template T
@@ -46,24 +47,9 @@ interface MockObject
     /**
      * @param Constraint|string $constraint
      *
-     * @return static
+     * @return InvocationMocker
      *
      * @throws RuntimeException
      */
     public function method($constraint);
-
-    /**
-     * @param mixed $value
-     * @param mixed $nextValues, ...
-     *
-     * @return static
-     */
-    public function willReturn($value, ...$nextValues);
-
-    /**
-     * @param Stub $stub
-     *
-     * @return InvocationMocker
-     */
-    public function will(Stub $stub);
 }