diff --git a/src/Bridges/NetteDI/OrmExtension.php b/src/Bridges/NetteDI/OrmExtension.php
index 095b396c6..6227f4772 100644
--- a/src/Bridges/NetteDI/OrmExtension.php
+++ b/src/Bridges/NetteDI/OrmExtension.php
@@ -177,6 +177,7 @@ protected function setupMetadataStorage(array $entityClassMap): void
 
 
 	/**
+	 * @param class-string<IModel> $modelClass
 	 * @param array{
 	 *     array<class-string<IRepository<IEntity>>, true>,
 	 *     array<string, class-string<IRepository<IEntity>>>,
diff --git a/src/Bridges/NetteDI/PhpDocRepositoryFinder.php b/src/Bridges/NetteDI/PhpDocRepositoryFinder.php
index 54c3df9fb..dd1738aa3 100644
--- a/src/Bridges/NetteDI/PhpDocRepositoryFinder.php
+++ b/src/Bridges/NetteDI/PhpDocRepositoryFinder.php
@@ -117,6 +117,9 @@ protected function setupMapperService(string $repositoryName, string $repository
 	}
 
 
+	/**
+	 * @param class-string<IRepository<IEntity>> $repositoryClass
+	 */
 	protected function setupRepositoryService(string $repositoryName, string $repositoryClass): void
 	{
 		$serviceName = $this->extension->prefix('repositories.' . $repositoryName);
diff --git a/tests/cases/integration/BridgeNetteDI/dic-extension-multiple.phpt b/tests/cases/integration/BridgeNetteDI/dic-extension-multiple.phpt
index 9d661e60a..e130334ec 100644
--- a/tests/cases/integration/BridgeNetteDI/dic-extension-multiple.phpt
+++ b/tests/cases/integration/BridgeNetteDI/dic-extension-multiple.phpt
@@ -23,10 +23,11 @@ function buildDic(string $config): Container
 	$cacheDir = TEMP_DIR . '/cache/bridge-nette-dic-extension-multiple';
 	$loader = new ContainerLoader($cacheDir);
 	$key = __FILE__ . ':' . __LINE__ . ':' . $config;
-	$className = $loader->load(function (Compiler $compiler) use ($config, $cacheDir): void {
+	$className = $loader->load(function (Compiler $compiler) use ($config, $cacheDir): ?string {
 		$compiler->addExtension('extensions', new ExtensionsExtension());
 		$compiler->addConfig(['parameters' => ['tempDir' => $cacheDir]]);
 		$compiler->loadConfig($config);
+		return null;
 	}, $key);
 
 	/** @var Container $dic */
diff --git a/tests/cases/integration/BridgeNetteDI/dic-extension-order.phpt b/tests/cases/integration/BridgeNetteDI/dic-extension-order.phpt
index 5672cfff3..f1dc5712c 100644
--- a/tests/cases/integration/BridgeNetteDI/dic-extension-order.phpt
+++ b/tests/cases/integration/BridgeNetteDI/dic-extension-order.phpt
@@ -19,10 +19,11 @@ function buildDic(string $config): Container
 	$cacheDir = TEMP_DIR . '/cache/bridge-nette-dic-extension-order';
 	$loader = new ContainerLoader($cacheDir);
 	$key = __FILE__ . ':' . __LINE__ . ':' . $config;
-	$className = $loader->load(function (Compiler $compiler) use ($config, $cacheDir): void {
+	$className = $loader->load(function (Compiler $compiler) use ($config, $cacheDir): ?string {
 		$compiler->addExtension('extensions', new ExtensionsExtension());
 		$compiler->addConfig(['parameters' => ['tempDir' => $cacheDir]]);
 		$compiler->loadConfig($config);
+		return null;
 	}, $key);
 
 	/** @var Container $dic */
diff --git a/tests/cases/integration/BridgeNetteDI/dic-finder.phpt b/tests/cases/integration/BridgeNetteDI/dic-finder.phpt
index 5cdd4d45d..3b7de0976 100644
--- a/tests/cases/integration/BridgeNetteDI/dic-finder.phpt
+++ b/tests/cases/integration/BridgeNetteDI/dic-finder.phpt
@@ -20,10 +20,11 @@ function buildDic(string $config): Container
 	$cacheDir = TEMP_DIR . '/cache/bridge-nette-di-dic-finder';
 	$loader = new ContainerLoader($cacheDir);
 	$key = __FILE__ . ':' . __LINE__ . ':' . $config;
-	$className = $loader->load(function (Compiler $compiler) use ($config, $cacheDir): void {
+	$className = $loader->load(function (Compiler $compiler) use ($config, $cacheDir): ?string {
 		$compiler->addExtension('extensions', new ExtensionsExtension());
 		$compiler->addConfig(['parameters' => ['tempDir' => $cacheDir]]);
 		$compiler->loadConfig($config);
+		return null;
 	}, $key);
 
 	/** @var Container $dic */