diff --git a/apps/tag/services/tag_distribution_services.py b/apps/tag/services/tag_distribution_services.py
index 2990481..2d9823f 100644
--- a/apps/tag/services/tag_distribution_services.py
+++ b/apps/tag/services/tag_distribution_services.py
@@ -79,7 +79,7 @@ class TagDistributionService:
 
         return {'tag_distributions': distributions, 'distributions_batch': distributions_batch}
 
-    def edit_distribution(self):
+    def edit_distribution(self, batch: TagDistributionBatch, data: dict = None):
         """
         edit record of distributed tags
         """
diff --git a/apps/tag/web/api/v1/serializers.py b/apps/tag/web/api/v1/serializers.py
index ef8226c..750049b 100644
--- a/apps/tag/web/api/v1/serializers.py
+++ b/apps/tag/web/api/v1/serializers.py
@@ -201,7 +201,7 @@ class TagDistributionBatchSerializer(serializers.ModelSerializer):
 
         representation['distributions'] = [{
             'id': dist.id,
-            'dist_identity': dist.identity,
+            'dist_identity': dist.dist_identity,
             'species_code': dist.species_code,
             'distributed_number': dist.distributed_number
         } for dist in instance.distributions.all()]