diff --git a/Database/Base/AbstractCollection.php b/Database/Base/AbstractCollection.php
index 79ee3503..d3b9135f 100644
--- a/Database/Base/AbstractCollection.php
+++ b/Database/Base/AbstractCollection.php
@@ -46,7 +46,7 @@ abstract class AbstractCollection extends Component implements \IteratorAggregat
 	 * @param array $array
 	 * @param null $model
 	 */
-	public function __construct($query, array $array = [], $model = null)
+	public function __construct(ActiveQuery $query, array $array = [], $model = null)
 	{
 		$this->_item = $array;
 		$this->query = $query;
diff --git a/Database/Base/CollectionIterator.php b/Database/Base/CollectionIterator.php
index 1b484820..d589e1c9 100644
--- a/Database/Base/CollectionIterator.php
+++ b/Database/Base/CollectionIterator.php
@@ -34,7 +34,7 @@ class CollectionIterator extends \ArrayIterator
 	 * @param int $flags
 	 * @throws Exception
 	 */
-	public function __construct($model, $query, $array = array(), $flags = 0)
+	public function __construct(ActiveRecord $model,ActiveQuery $query, $array = array(), $flags = 0)
 	{
 		$this->model = $model;
 		$this->query = $query;