diff --git a/src/Aggregation/AbstractAggregation.php b/src/Aggregation/AbstractAggregation.php
index 03b369415a2b1d2ea37fb20bf04f5ebfbc3f55eb..32977f49cd33dbdd3bcf33456c032674f3bb0965 100644
--- a/src/Aggregation/AbstractAggregation.php
+++ b/src/Aggregation/AbstractAggregation.php
@@ -93,7 +93,7 @@ abstract class AbstractAggregation implements BuilderInterface
     /**
      * Returns all sub aggregations.
      *
-     * @return BuilderBag[]
+     * @return BuilderBag[]|BuilderInterface[]
      */
     public function getAggregations()
     {
@@ -108,7 +108,7 @@ abstract class AbstractAggregation implements BuilderInterface
      * Returns sub aggregation.
      * @param string $name Aggregation name to return.
      *
-     * @return AbstractAggregation|null
+     * @return AbstractAggregation|BuilderInterface|null
      */
     public function getAggregation($name)
     {