diff --git a/src/CacheTrait.php b/src/CacheTrait.php
index e217440..d960278 100644
--- a/src/CacheTrait.php
+++ b/src/CacheTrait.php
@@ -59,7 +59,7 @@ trait CacheTrait
         $this->eventRegister(new class($this->cache, $this->entityCacheKey()) implements Event\Query\Update {
             use Cache\CacheEventTrait;
 
-            public function execute(RepositoryInterface $repository, object|array $entity, ?array $dataset = null, bool $replace = false): void
+            public function execute(RepositoryInterface $repository, object|array $entity, array $dataset, array $diff, bool $replace = false): void
             {
                 $this->purgeCache();
             }