diff --git a/tests/unit/framework/data/ActiveDataProviderTest.php b/tests/unit/framework/data/ActiveDataProviderTest.php
index 608fed6..7df2983 100644
--- a/tests/unit/framework/data/ActiveDataProviderTest.php
+++ b/tests/unit/framework/data/ActiveDataProviderTest.php
@@ -49,6 +49,7 @@ class ActiveDataProviderTest extends DatabaseTestCase
 	{
 		$query = new Query;
 		$provider = new ActiveDataProvider(array(
+			'db' => $this->getConnection(),
 			'query' => $query->from('tbl_order')->orderBy('id'),
 		));
 		$orders = $provider->getItems();
@@ -58,6 +59,7 @@ class ActiveDataProviderTest extends DatabaseTestCase
 
 		$query = new Query;
 		$provider = new ActiveDataProvider(array(
+			'db' => $this->getConnection(),
 			'query' => $query->from('tbl_order'),
 			'pagination' => array(
 				'pageSize' => 2,