diff --git a/src/Aggregation/TopHitsAggregation.php b/src/Aggregation/TopHitsAggregation.php
index 0212be360f0d7f693a3b93c71726a44ccf83d738..d41e3d82bab0599a5bf40253c4a63283e63a0edf 100644
--- a/src/Aggregation/TopHitsAggregation.php
+++ b/src/Aggregation/TopHitsAggregation.php
@@ -12,7 +12,7 @@
 namespace ONGR\ElasticsearchDSL\Aggregation;
 
 use ONGR\ElasticsearchDSL\Aggregation\Type\MetricTrait;
-use ONGR\ElasticsearchDSL\Sort\Sorts;
+use ONGR\ElasticsearchDSL\BuilderInterface;
 
 /**
  * Top hits aggregation.
@@ -32,7 +32,7 @@ class TopHitsAggregation extends AbstractAggregation
     private $from;
 
     /**
-     * @var Sorts How the top matching hits should be sorted.
+     * @var BuilderInterface How the top matching hits should be sorted.
      */
     private $sort;
 
@@ -42,7 +42,7 @@ class TopHitsAggregation extends AbstractAggregation
      * @param string     $name Aggregation name.
      * @param null|int   $size Number of top matching hits to return per bucket.
      * @param null|int   $from The offset from the first result you want to fetch.
-     * @param null|Sorts $sort How the top matching hits should be sorted.
+     * @param null|BuilderInterface $sort How the top matching hits should be sorted.
      */
     public function __construct($name, $size = null, $from = null, $sort = null)
     {
@@ -75,7 +75,7 @@ class TopHitsAggregation extends AbstractAggregation
     /**
      * Return sort.
      *
-     * @return Sorts
+     * @return BuilderInterface
      */
     public function getSort()
     {
@@ -85,7 +85,7 @@ class TopHitsAggregation extends AbstractAggregation
     /**
      * Set sort.
      *
-     * @param Sorts $sort
+     * @param BuilderInterface $sort
      */
     public function setSort($sort)
     {
@@ -142,7 +142,7 @@ class TopHitsAggregation extends AbstractAggregation
      */
     private function getFilteredData()
     {
-        $fd = array_filter(
+        $output = array_filter(
             [
                 'sort' => $this->getSort() ? $this->getSort()->toArray() : [],
                 'size' => $this->getSize(),
@@ -153,6 +153,6 @@ class TopHitsAggregation extends AbstractAggregation
             }
         );
 
-        return $this->processArray($fd);
+        return $output;
     }
 }
diff --git a/tests/Aggregation/TopHitsAggregationTest.php b/tests/Aggregation/TopHitsAggregationTest.php
index fa2e91ff9e3bd4416d4e838e2d8ecfb08520889a..cd63b502d2b0932580cc935b5e917ce02cb467e9 100644
--- a/tests/Aggregation/TopHitsAggregationTest.php
+++ b/tests/Aggregation/TopHitsAggregationTest.php
@@ -40,23 +40,4 @@ class TopHitsAggregationTest extends \PHPUnit_Framework_TestCase
 
         $this->assertEquals($expected, $aggregation->toArray());
     }
-
-    /**
-     * Tests if adding parameters has any effect.
-     */
-    public function testParameters()
-    {
-        $topHitsAggregation = new TopHitsAggregation('test');
-        $topHitsAggregation->addParameter('_source', ['include' => ['title']]);
-        $expectedAgg = new \stdClass();
-        $expectedAgg->sort = [];
-        $expectedAgg->_source = ['include' => ['title']];
-        $expected = [
-            'agg_test' => [
-                'top_hits' => $expectedAgg,
-            ],
-        ];
-
-        $this->assertEquals($expected, $topHitsAggregation->toArray());
-    }
 }