diff --git a/src/Connection/Manager.php b/src/Connection/Manager.php
index 96779bb9..848350dc 100644
--- a/src/Connection/Manager.php
+++ b/src/Connection/Manager.php
@@ -46,12 +46,12 @@ class Manager
      * pure PHP. After adding a connection into the Manager you are ready to
      * persist and query your models.
      *
-     * @param IlluminateContainer $connection connection instance to be used in database interactions
+     * @param Connection $connection connection instance to be used in database interactions
      */
-    public function setConnection(IlluminateContainer $connection): bool
+    public function setConnection(Connection $connection): bool
     {
         $this->init();
-        $this->container->instance(IlluminateContainer::class, $this->connection);
+        $this->container->instance(Connection::class, $this->connection);
 
         $this->connection = $connection;
 
diff --git a/tests/Unit/Connection/ManagerTest.php b/tests/Unit/Connection/ManagerTest.php
index ed521de3..30475799 100644
--- a/tests/Unit/Connection/ManagerTest.php
+++ b/tests/Unit/Connection/ManagerTest.php
@@ -14,7 +14,7 @@ public function testShouldAddAndGetConnection(): void
     {
         // Set
         $manager = new Manager();
-        $connection = m::mock(IlluminateContainer::class);
+        $connection = m::mock(Connection::class);
         $client = m::mock(Client::class);
 
         // Expectations