diff --git a/src/Kdyby/RabbitMq/Diagnostics/Panel.php b/src/Kdyby/RabbitMq/Diagnostics/Panel.php
index 0d88ccd8..ff506951 100644
--- a/src/Kdyby/RabbitMq/Diagnostics/Panel.php
+++ b/src/Kdyby/RabbitMq/Diagnostics/Panel.php
@@ -10,7 +10,7 @@
 
 namespace Kdyby\RabbitMq\Diagnostics;
 
-use Kdyby\RabbitMq\Connection;
+use Kdyby\RabbitMq\IConnection;
 use Nette;
 use Nette\Utils\Html;
 use Tracy\Debugger;
@@ -137,10 +137,10 @@ public function published($message)
 
 
 	/**
-	 * @param Connection $connection
+	 * @param IConnection $connection
 	 * @return Panel
 	 */
-	public function register(Connection $connection)
+	public function register(IConnection $connection)
 	{
 		Debugger::getBar()->addPanel($this);
 		return $this;