diff --git a/composer.json b/composer.json
index 0404969..d84633a 100644
--- a/composer.json
+++ b/composer.json
@@ -10,7 +10,6 @@
         }
     },
     "require": {
-        "codeception/phpunit-wrapper": "^8.0.4",
         "phpunit/phpunit": "^8.4"
     }
 }
diff --git a/tests/StubTest.php b/tests/StubTest.php
index 3a46157..0afb2a6 100644
--- a/tests/StubTest.php
+++ b/tests/StubTest.php
@@ -2,7 +2,7 @@
 require_once __DIR__ .'/ResetMocks.php';
 use Codeception\Stub;
 
-class StubTest extends \Codeception\PHPUnit\TestCase
+class StubTest extends \PHPUnit\Framework\TestCase
 {
     use ResetMocks;
     /**
@@ -10,7 +10,7 @@ class StubTest extends \Codeception\PHPUnit\TestCase
      */
     protected $dummy;
 
-    public function _setUp()
+    public function setUp(): void
     {
         require_once $file = __DIR__. '/_data/DummyOverloadableClass.php';
         require_once $file = __DIR__. '/_data/DummyClass.php';
diff --git a/tests/StubTraitTest.php b/tests/StubTraitTest.php
index 1c0be97..5be5790 100644
--- a/tests/StubTraitTest.php
+++ b/tests/StubTraitTest.php
@@ -1,7 +1,7 @@
 <?php
 require_once __DIR__ .'/ResetMocks.php';
 
-class StubTraitTest extends \Codeception\PHPUnit\TestCase
+class StubTraitTest extends \PHPUnit\Framework\TestCase
 {
     use ResetMocks;
     use \Codeception\Test\Feature\Stub;
@@ -10,7 +10,7 @@ class StubTraitTest extends \Codeception\PHPUnit\TestCase
      */
     protected $dummy;
 
-    public function _setUp()
+    public function setUp(): void
     {
         require_once $file = __DIR__. '/_data/DummyOverloadableClass.php';
         require_once $file = __DIR__. '/_data/DummyClass.php';