diff --git a/tests/KdybyTests/Redis/RedisSessionHandlerTest.phpt b/tests/KdybyTests/Redis/RedisSessionHandlerTest.phpt
index b93fb90..bd3a8bb 100644
--- a/tests/KdybyTests/Redis/RedisSessionHandlerTest.phpt
+++ b/tests/KdybyTests/Redis/RedisSessionHandlerTest.phpt
@@ -273,7 +273,7 @@ class RedisSessionHandlerTest extends \KdybyTests\Redis\AbstractRedisTestCase
 				[],
 				'GET',
 			];
-			if (\class_exists(\Nette\DI\Definitions\ServiceDefinition::class)) {
+			if (\class_exists(\Nette\Http\UrlImmutable::class)) {
 				unset($arguments[1]);
 			}
 
@@ -336,7 +336,7 @@ class RedisSessionHandlerTest extends \KdybyTests\Redis\AbstractRedisTestCase
 			[],
 			'GET',
 		];
-		if (\class_exists(\Nette\DI\Definitions\ServiceDefinition::class)) {
+		if (\class_exists(\Nette\Http\UrlImmutable::class)) {
 			unset($arguments[1]);
 		}