diff --git a/src/DI/TuyaExtension.php b/src/DI/TuyaExtension.php
index 394aa5b..5674b06 100644
--- a/src/DI/TuyaExtension.php
+++ b/src/DI/TuyaExtension.php
@@ -35,6 +35,7 @@
 use FastyBird\Library\Metadata;
 use FastyBird\Library\Metadata\Documents as MetadataDocuments;
 use FastyBird\Module\Devices\DI as DevicesDI;
+use Nette\Bootstrap;
 use Nette\DI;
 use Nettrine\ORM as NettrineORM;
 use function array_keys;
@@ -60,7 +61,7 @@ public static function register(
 	): void
 	{
 		$config->onCompile[] = static function (
-			ApplicationBoot\Configurator $config,
+			Bootstrap\Configurator $config,
 			DI\Compiler $compiler,
 		) use ($extensionName): void {
 			$compiler->addExtension($extensionName, new self());