diff --git a/sites/all/modules/figli/edlp_corpus/edlp_corpus.module b/sites/all/modules/figli/edlp_corpus/edlp_corpus.module
index 574d0b32b..fa6eea02b 100644
--- a/sites/all/modules/figli/edlp_corpus/edlp_corpus.module
+++ b/sites/all/modules/figli/edlp_corpus/edlp_corpus.module
@@ -10,6 +10,8 @@
 use Drupal\Core\Entity\EntityInterface;
 use Drupal\Core\Entity\Display\EntityViewDisplayInterface;
 use Drupal\Core\Url;
+use Drupal\workflow\Entity\WorkflowManager;
+
 
 /**
  * Implements hook_theme().
@@ -175,7 +177,7 @@ function edlp_corpus_node_create(EntityInterface $node) {
  * @see hook_entity_update()
  */
 function edlp_corpus_node_update(EntityInterface $node) {
-  _edlp_corpus_invalidate_corpus_cache();
+  _edlp_corpus_invalidate_corpus_cache($node);
 }
 /**
  * Respond to entity deletion.
@@ -186,5 +188,5 @@ function edlp_corpus_node_update(EntityInterface $node) {
  * @see hook_ENTITY_TYPE_delete()
  */
 function edlp_corpus_node_delete(EntityInterface $node) {
-  _edlp_corpus_invalidate_corpus_cache();
+  _edlp_corpus_invalidate_corpus_cache($node);
 }