diff --git a/lib/Doctrine/ODM/MongoDB/Aggregation/Stage/AddFields.php b/lib/Doctrine/ODM/MongoDB/Aggregation/Stage/AddFields.php
index b658656bb..61355728b 100644
--- a/lib/Doctrine/ODM/MongoDB/Aggregation/Stage/AddFields.php
+++ b/lib/Doctrine/ODM/MongoDB/Aggregation/Stage/AddFields.php
@@ -11,8 +11,9 @@
  *
  * @psalm-import-type OperatorExpression from Expr
  * @psalm-type AddFieldsStageExpression = array{'$addFields': array<string, OperatorExpression|mixed>}
+ * @final
  */
-final class AddFields extends Operator
+class AddFields extends Operator
 {
     /** @return AddFieldsStageExpression */
     public function getExpression(): array
diff --git a/lib/Doctrine/ODM/MongoDB/Aggregation/Stage/Set.php b/lib/Doctrine/ODM/MongoDB/Aggregation/Stage/Set.php
index cca5aed34..3d4afacac 100644
--- a/lib/Doctrine/ODM/MongoDB/Aggregation/Stage/Set.php
+++ b/lib/Doctrine/ODM/MongoDB/Aggregation/Stage/Set.php
@@ -11,8 +11,9 @@
  *
  * @psalm-import-type OperatorExpression from Expr
  * @psalm-type SetStageExpression = array{'$set': array<string, OperatorExpression|mixed>}
+ * @final
  */
-final class Set extends Operator
+class Set extends Operator
 {
     /** @psalm-return SetStageExpression */
     public function getExpression(): array