diff --git a/edivorce/apps/core/models.py b/edivorce/apps/core/models.py
index 37255259..95933c4b 100644
--- a/edivorce/apps/core/models.py
+++ b/edivorce/apps/core/models.py
@@ -112,9 +112,6 @@ class UserResponse(models.Model):
 
 
 class Document(models.Model):
-    """
-    This is only a POC model and should not be loaded on a production system.
-    """
     filename = models.CharField(max_length=128, null=True)  # saving the original filename separately
     """ File name and extension """
 
diff --git a/edivorce/apps/core/serializer.py b/edivorce/apps/core/serializer.py
index 4d631bf5..12981ab5 100644
--- a/edivorce/apps/core/serializer.py
+++ b/edivorce/apps/core/serializer.py
@@ -25,7 +25,7 @@ class UserResponseSerializer(serializers.ModelSerializer):
 
 class DocumentSerializer(serializers.ModelSerializer):
     doc_type = serializers.CharField()
-    party_code = serializers.IntegerField()
+    party_code = serializers.IntegerField(min_value=0, max_value=2)
     file = serializers.FileField()
     filename = serializers.CharField(read_only=True)
     size = serializers.IntegerField(read_only=True)