diff --git a/src/Bundle/ChillBudgetBundle/Calculator/CalculatorInterface.php b/src/Bundle/ChillBudgetBundle/Calculator/CalculatorInterface.php
index 63b4c7294..9acb2ff86 100644
--- a/src/Bundle/ChillBudgetBundle/Calculator/CalculatorInterface.php
+++ b/src/Bundle/ChillBudgetBundle/Calculator/CalculatorInterface.php
@@ -12,11 +12,13 @@ declare(strict_types=1);
 namespace Chill\BudgetBundle\Calculator;
 
 use Chill\BudgetBundle\Entity\AbstractElement;
+use Chill\BudgetBundle\Entity\Charge;
+use Chill\BudgetBundle\Entity\Resource;
 
 interface CalculatorInterface
 {
     /**
-     * @param AbstractElement[] $elements
+     * @param array<Charge|Resource> $elements
      */
     public function calculate(array $elements): ?CalculatorResult;