diff --git a/src/Engines/TNTSearchEngine.php b/src/Engines/TNTSearchEngine.php
index b6e5d70..81d733a 100644
--- a/src/Engines/TNTSearchEngine.php
+++ b/src/Engines/TNTSearchEngine.php
@@ -218,11 +218,11 @@ public function map(Builder $builder, $results, $model)
         }
 
         // sort models by tnt search result set
-        return $model->newCollection($results['ids'])->map(function ($hit, $key) use ($models, $results) {
+        return $model->newCollection(collect($results['ids'])->map(function ($hit) use ($models, $results) {
             if (isset($models[$hit])) {
                 return $models[$hit]->setAttribute('__tntSearchScore__', $results['docScores'][$hit]);
             }
-        })->filter()->values();
+        })->filter()->all());
     }
 
     /**