diff --git a/Repository/EtlExecutionRepository.php b/Repository/EtlExecutionRepository.php
index 34d9e71..2651a64 100644
--- a/Repository/EtlExecutionRepository.php
+++ b/Repository/EtlExecutionRepository.php
@@ -32,13 +32,13 @@ public function updateStepStats(EtlExecution $execution, string $stepStats): voi
         /** @var \Doctrine\DBAL\Connection $connection */
         $connection = $this->registry->getConnection();
 
-        $query = $connection->createQueryBuilder()
-            ->update('EtlExecution', 'e')
+        $query = $this->createQueryBuilder('e')
+            ->update()
             ->set('e.stepStats', ':stepStats')
             ->where('e.id = :executionId')
             ->setParameter('stepStats', $stepStats)
             ->setParameter('executionId', $execution->getId());
-        $query->executeQuery();
+        $query->execute();
 
         if ($connection->getTransactionNestingLevel() > 0) {
             $connection->commit();
diff --git a/Services/ChainProcessorsManager.php b/Services/ChainProcessorsManager.php
index a352ec9..63a42d2 100644
--- a/Services/ChainProcessorsManager.php
+++ b/Services/ChainProcessorsManager.php
@@ -106,7 +106,7 @@ public function execute(string $chainName, iterable $iterator, array $params,  ?
      * Execute a chain from it's entity.
      *
      */
-    public function executeFromEtlEntity(EtlExecution $execution, iterable $iterator = null, ?callable $observerCallback = null): void
+    public function executeFromEtlEntity(EtlExecution $execution, ?iterable $iterator = null, ?callable $observerCallback = null): void
     {
         $chainName = $execution->getName();
         $logger = $this->loggerFactory->get($execution);