diff --git a/app/Http/Middleware/SetDocumentDb.php b/app/Http/Middleware/SetDocumentDb.php
index 83d0d0be88bd..8559c667d931 100644
--- a/app/Http/Middleware/SetDocumentDb.php
+++ b/app/Http/Middleware/SetDocumentDb.php
@@ -32,9 +32,9 @@ class SetDocumentDb
             'errors' => new stdClass,
         ];
 
-        if ($request->has('document_hash') && config('ninja.db.multi_db_enabled')) {
+        if (config('ninja.db.multi_db_enabled')) {
             
-            if (! MultiDB::documentFindAndSetDb($request->input('document_hash'))) 
+            if (! MultiDB::documentFindAndSetDb($request->segment(2))) 
                 return response()->json($error, 400);
             
         }
diff --git a/app/Libraries/MultiDB.php b/app/Libraries/MultiDB.php
index 8120da08925d..495b55f97fa6 100644
--- a/app/Libraries/MultiDB.php
+++ b/app/Libraries/MultiDB.php
@@ -15,6 +15,7 @@ use App\Models\Client;
 use App\Models\ClientContact;
 use App\Models\Company;
 use App\Models\CompanyToken;
+use App\Models\Document;
 use App\Models\User;
 use Illuminate\Support\Facades\DB;
 use Illuminate\Support\Str;