diff --git a/composer.json b/composer.json
index d1b82bc73..a41c584e5 100644
--- a/composer.json
+++ b/composer.json
@@ -26,7 +26,7 @@
 		"latte/latte": "Allows using Latte in templates"
 	},
 	"require-dev": {
-		"nette/tester": "^2.5",
+		"nette/tester": "^2.5.2",
 		"nette/di": "^3.2",
 		"nette/forms": "^3.2",
 		"nette/robot-loader": "^4.0",
diff --git a/tests/bootstrap.php b/tests/bootstrap.php
index 9b8053267..aeff549f3 100644
--- a/tests/bootstrap.php
+++ b/tests/bootstrap.php
@@ -13,6 +13,7 @@
 
 // configure environment
 Tester\Environment::setup();
+Tester\Environment::setupFunctions();
 date_default_timezone_set('Europe/Prague');
 Mockery::setLoader(new Mockery\Loader\RequireLoader(getTempDir()));
 
@@ -44,11 +45,9 @@ function getTempDir(): string
 }
 
 
-function test(string $title, Closure $function): void
-{
-	$function();
+tearDown(function () {
 	Mockery::close();
-}
+});
 
 
 class Notes