diff --git a/templates/module/src/Entity/entity-content.php.twig b/templates/module/src/Entity/entity-content.php.twig
index b2fc784b8..37f2b3d8b 100644
--- a/templates/module/src/Entity/entity-content.php.twig
+++ b/templates/module/src/Entity/entity-content.php.twig
@@ -9,7 +9,7 @@ namespace Drupal\{{ module }}\Entity;
 {% endblock %}
 
 {% block use_class %}
-{% if has_owner %}
+{% if has_owner or revisionable%}
 use Drupal\Core\Entity\EntityStorageInterface;
 {% endif %}
 use Drupal\Core\Field\BaseFieldDefinition;