diff --git a/tests/unit/data/ar/Order.php b/tests/unit/data/ar/Order.php
index fb302be40b..659f82201f 100644
--- a/tests/unit/data/ar/Order.php
+++ b/tests/unit/data/ar/Order.php
@@ -118,4 +118,12 @@ class Order extends ActiveRecord
             return false;
         }
     }
+
+    public function attributeLabels()
+    {
+        return [
+            'customer_id' => 'Customer',
+            'total' => 'Invoice Total',
+        ];
+    }
 }
diff --git a/tests/unit/framework/widgets/LinkSorterTest.php b/tests/unit/framework/widgets/LinkSorterTest.php
index 7ea661934d..01dc32e272 100644
--- a/tests/unit/framework/widgets/LinkSorterTest.php
+++ b/tests/unit/framework/widgets/LinkSorterTest.php
@@ -9,16 +9,21 @@ use yii\db\Query;
 use yii\widgets\Breadcrumbs;
 use yii\widgets\LinkSorter;
 use yii\widgets\ListView;
-use yiiunit\data\ar\elasticsearch\Order;
+use yiiunit\data\ar\ActiveRecord;
+use yiiunit\data\ar\Order;
+use yiiunit\framework\db\DatabaseTestCase;
 
 /**
  * @group widgets
  */
-class LinkSorterTest extends \yiiunit\TestCase
+class LinkSorterTest extends DatabaseTestCase
 {
+    protected $driverName = 'sqlite';
+
     protected function setUp()
     {
         parent::setUp();
+        ActiveRecord::$db = $this->getConnection();
         $this->mockWebApplication();
         $this->breadcrumbs = new Breadcrumbs();
     }