diff --git a/src/Eloquent/Docs/ModelDocs.php b/src/Eloquent/Docs/ModelDocs.php
index 74f44f7..bcef48f 100644
--- a/src/Eloquent/Docs/ModelDocs.php
+++ b/src/Eloquent/Docs/ModelDocs.php
@@ -9,6 +9,7 @@
 use Illuminate\Contracts\Pagination\LengthAwarePaginator;
 use Illuminate\Pagination\Cursor;
 use PDPhilip\Elasticsearch\Collection\ElasticCollection;
+use PDPhilip\Elasticsearch\Connection;
 use PDPhilip\Elasticsearch\Eloquent\Model;
 use PDPhilip\Elasticsearch\Pagination\SearchAfterPaginator;
 use PDPhilip\Elasticsearch\Query\Builder;
@@ -93,7 +94,7 @@
  *
  * Index Methods ---------------------------------
  * @method static string getQualifiedKeyName()
- * @method static string getConnection()
+ * @method static Connection getConnection()
  * @method static void truncate()
  * @method static bool indexExists()
  * @method static bool deleteIndexIfExists()