diff --git a/apps/tag/models.py b/apps/tag/models.py
index 58fb659..f1bcee9 100644
--- a/apps/tag/models.py
+++ b/apps/tag/models.py
@@ -138,6 +138,12 @@ class TagDistributionBatch(BaseModel):
         null=True
     )
     dist_batch_identity = models.CharField(max_length=20, default="0", unique=True, null=True)
+    owner_org = models.ForeignKey(
+        Organization,
+        on_delete=models.CASCADE,
+        related_name='tag_distribution_batch',
+        null=True
+    )
     assigner_org = models.ForeignKey(
         Organization,
         on_delete=models.CASCADE,
@@ -163,6 +169,7 @@ class TagDistributionBatch(BaseModel):
     total_tag_count = models.IntegerField(default=0)
     total_distributed_tag_count = models.PositiveBigIntegerField(default=0)
     remaining_tag_count = models.PositiveBigIntegerField(default=0)
+    top_root_distribution = models.BooleanField(default=False)
     is_closed = models.BooleanField(default=False)
 
     def __str__(self):