diff --git a/app/models/solver_run.py b/app/models/solver_run.py
index a9a6d00..77df041 100644
--- a/app/models/solver_run.py
+++ b/app/models/solver_run.py
@@ -69,7 +69,7 @@ class SolverRun(BaseModel):
             if bundle.id == bundle_id:
                 return bundle
 
-    def get_constraint_by_type(self, type: str) -> Constraint or None:
+    def get_constraint_by_type(self, type: str) -> GenericConstraint|MetadataConstraint|BundleConstraint|FormUniquenessConstraint|TotalFormItemsConstraint or None:
         for constraint in self.constraints:
             if type == constraint.reference_attribute.type:
                 return constraint
@@ -130,7 +130,7 @@ class SolverRun(BaseModel):
 
         logging.info('Bundles Generated...')
 
-    def get_constraint(self, name: str) -> Constraint:
+    def get_constraint(self, name: str) -> GenericConstraint|MetadataConstraint|BundleConstraint|FormUniquenessConstraint|TotalFormItemsConstraint:
         return next((constraint for constraint in self.constraints
                      if constraint.reference_attribute.id == name), None)