diff --git a/composer.json b/composer.json
index 1a062ab..6cc6cca 100644
--- a/composer.json
+++ b/composer.json
@@ -15,7 +15,7 @@
     ],
     "require": {
         "mockery/mockery": "^0.8|^0.9|^1.0",
-        "codeception/codeception": "^2.0|^3.0|^4.0"
+        "codeception/codeception": "^2.4|^3.0|^4.0"
     },
     "autoload": {
         "psr-4": {
diff --git a/src/MockeryModule/Test.php b/src/MockeryModule/Test.php
index 03c613c..a6d4d9f 100644
--- a/src/MockeryModule/Test.php
+++ b/src/MockeryModule/Test.php
@@ -2,7 +2,7 @@
 
 namespace Codeception\MockeryModule;
 
-use Codeception\TestCase\Test as BaseTest;
+use Codeception\Test\Unit as BaseTest;
 use Mockery;
 
 /**
@@ -16,9 +16,9 @@
  */
 class Test extends BaseTest
 {
-    protected function tearDown()
+    protected function _tearDown()
     {
-        parent::tearDown();
+        parent::_tearDown();
         Mockery::close();
     }
 }