diff --git a/src/Repository.php b/src/Repository.php
index fa2709d..1c724e0 100644
--- a/src/Repository.php
+++ b/src/Repository.php
@@ -2,6 +2,7 @@
 
 namespace Optimus\Database\Eloquent;
 
+use Illuminate\Database\DatabaseManager;
 use Illuminate\Database\Eloquent\Builder;
 use Illuminate\Database\Eloquent\Collection;
 use Optimus\Database\Eloquent\Model;
@@ -11,6 +12,8 @@ abstract class Repository
 {
     use EloquentBuilderTrait;
 
+    protected $database;
+
     protected $model;
 
     protected $sortProperty = null;
@@ -20,8 +23,9 @@ abstract class Repository
 
     abstract protected function getModel();
 
-    final public function __construct()
+    final public function __construct(DatabaseManager $database)
     {
+        $this->database = $database;
         $this->model = $this->getModel();
     }