diff --git a/src/DataSources/DibiFluent.php b/src/DataSources/DibiFluent.php
index 04311cc7..dce73bdb 100644
--- a/src/DataSources/DibiFluent.php
+++ b/src/DataSources/DibiFluent.php
@@ -44,7 +44,7 @@ class DibiFluent implements IDataSource
     /**
      * @param \DibiFluent $fluent
      */
-    public function __construct(\DibiFluent $fluent)
+    public function __construct(\Dibi\Fluent $fluent)
     {
         $this->fluent = $fluent;
     }
diff --git a/src/DataSources/Model.php b/src/DataSources/Model.php
index 8e053856..4d4472dd 100644
--- a/src/DataSources/Model.php
+++ b/src/DataSources/Model.php
@@ -40,7 +40,7 @@ class Model
      */
     public function __construct($model)
     {
-        if ($model instanceof \DibiFluent) {
+        if ($model instanceof \Dibi\Fluent) {
             $dataSource = new DibiFluent($model);
         } elseif ($model instanceof \Nette\Database\Table\Selection) {
             $dataSource = new NetteDatabase($model);