diff --git a/src/Plugin/Block/EntityView.php b/src/Plugin/Block/EntityView.php
index 51342d0..f808efe 100644
--- a/src/Plugin/Block/EntityView.php
+++ b/src/Plugin/Block/EntityView.php
@@ -96,8 +96,7 @@ class EntityView extends BlockBase implements ContextAwarePluginInterface, Conta
     $view_builder = $this->entityManager->getViewBuilder($entity->getEntityTypeId());
     $build = $view_builder->view($entity, $this->configuration['view_mode']);
 
-    CacheableMetadata::createFromObject($this->getContext('entity'))
-      ->applyTo($build);
+    CacheableMetadata::createFromObject($entity)->applyTo($build);
 
     return $build;
   }