diff --git a/src/Entity/ReplicationLog.php b/src/Entity/ReplicationLog.php
index 725384e..fb2af3f 100644
--- a/src/Entity/ReplicationLog.php
+++ b/src/Entity/ReplicationLog.php
@@ -2,7 +2,6 @@
 
 namespace Drupal\replication\Entity;
 
-use Drupal\Core\Entity\Entity;
 use Drupal\Core\Entity\ContentEntityBase;
 use Drupal\Core\Entity\EntityTypeInterface;
 use Drupal\Core\Field\BaseFieldDefinition;
@@ -19,6 +18,7 @@ use Drupal\Core\Field\FieldStorageDefinitionInterface;
  *   },
  *   base_table = "replication_log",
  *   revision_table = "replication_log_revision",
+ *   revision_data_table = "replication_log_field_revision",
  *   fieldable = FALSE,
  *   entity_keys = {
  *     "id" = "id",