diff --git a/composer.json b/composer.json
index ff3fac1e..5628f8f9 100644
--- a/composer.json
+++ b/composer.json
@@ -21,7 +21,7 @@
 	},
 	"require-dev": {
 		"nette/tester": "^2.0",
-		"nette/di": "^2.4 || ~3.0.0",
+		"nette/di": "^v3.0.0-beta2",
 		"latte/latte": "^2.4",
 		"tracy/tracy": "^2.4"
 	},
diff --git a/src/Bridges/CacheDI/CacheExtension.php b/src/Bridges/CacheDI/CacheExtension.php
index b85fd60c..3a18223b 100644
--- a/src/Bridges/CacheDI/CacheExtension.php
+++ b/src/Bridges/CacheDI/CacheExtension.php
@@ -39,12 +39,12 @@ public function loadConfiguration()
 
 		if (extension_loaded('pdo_sqlite')) {
 			$builder->addDefinition($this->prefix('journal'))
-				->setClass(Nette\Caching\Storages\IJournal::class)
+				->setType(Nette\Caching\Storages\IJournal::class)
 				->setFactory(Nette\Caching\Storages\SQLiteJournal::class, [$this->tempDir . '/cache/journal.s3db']);
 		}
 
 		$builder->addDefinition($this->prefix('storage'))
-			->setClass(Nette\Caching\IStorage::class)
+			->setType(Nette\Caching\IStorage::class)
 			->setFactory(Nette\Caching\Storages\FileStorage::class, [$this->tempDir . '/cache']);
 
 		if ($this->name === 'cache') {