diff --git a/src/DynamoDbQueryBuilder.php b/src/DynamoDbQueryBuilder.php
index 77f9d4c..a0b0f73 100644
--- a/src/DynamoDbQueryBuilder.php
+++ b/src/DynamoDbQueryBuilder.php
@@ -526,7 +526,7 @@ protected function getAll($columns = [], $limit = -1, $use_iterator = true)
             if ($this->conditionsAreExactSearch()) {
                 $item = $this->find($conditionValue, $columns);
 
-                return new Collection([$item]);
+                return $this->getModel()->newCollection([$item]);
             }
         }
 
@@ -579,7 +579,7 @@ protected function getAll($columns = [], $limit = -1, $use_iterator = true)
             $results[] = $model;
         }
 
-        return new Collection($results);
+        return $this->getModel()->newCollection($results);
     }
 
     protected function buildExpressionQuery()