diff --git a/src/DI/WsServerExtension.php b/src/DI/WsServerExtension.php
index fb29bbe..1e5d0d6 100644
--- a/src/DI/WsServerExtension.php
+++ b/src/DI/WsServerExtension.php
@@ -23,6 +23,7 @@
 use FastyBird\Plugin\WsServer\Subscribers;
 use IPub\WebSockets;
 use Nette;
+use Nette\Bootstrap;
 use Nette\DI;
 use Nette\PhpGenerator;
 use Nette\Schema;
@@ -51,7 +52,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());