diff --git a/metadata-ingestion/src/datahub/api/entities/assertion/field_assertion.py b/metadata-ingestion/src/datahub/api/entities/assertion/field_assertion.py
index ae062c3a8e..9d6a990b95 100644
--- a/metadata-ingestion/src/datahub/api/entities/assertion/field_assertion.py
+++ b/metadata-ingestion/src/datahub/api/entities/assertion/field_assertion.py
@@ -38,7 +38,7 @@ class FieldValuesFailThreshold(v1_ConfigModel):
         return FieldValuesFailThresholdClass(
             type=(
                 FieldValuesFailThresholdTypeClass.COUNT
-                if self.type == Literal["count"]
+                if self.type == "count"
                 else FieldValuesFailThresholdTypeClass.PERCENTAGE
             ),
             value=self.value,
@@ -81,7 +81,7 @@ class FieldValuesAssertion(BaseEntityAssertion):
                     excludeNulls=self.exclude_nulls,
                     transform=(
                         FieldTransformClass(type=FieldTransformTypeClass.LENGTH)
-                        if self.field_transform == Literal["length"]
+                        if self.field_transform == FieldTransform.LENGTH
                         else None
                     ),
                 ),
diff --git a/metadata-ingestion/src/datahub/api/entities/assertion/sql_assertion.py b/metadata-ingestion/src/datahub/api/entities/assertion/sql_assertion.py
index 3d12cfde42..625710cbfe 100644
--- a/metadata-ingestion/src/datahub/api/entities/assertion/sql_assertion.py
+++ b/metadata-ingestion/src/datahub/api/entities/assertion/sql_assertion.py
@@ -58,7 +58,7 @@ class SqlMetricChangeAssertion(BaseEntityAssertion):
                 statement=self.statement,
                 changeType=(
                     AssertionValueChangeType.ABSOLUTE
-                    if self.change_type == Literal["absolute"]
+                    if self.change_type == "absolute"
                     else AssertionValueChangeType.PERCENTAGE
                 ),
                 operator=self.operator.operator,
diff --git a/metadata-ingestion/src/datahub/api/entities/assertion/volume_assertion.py b/metadata-ingestion/src/datahub/api/entities/assertion/volume_assertion.py
index da6a125874..8254ac46ae 100644
--- a/metadata-ingestion/src/datahub/api/entities/assertion/volume_assertion.py
+++ b/metadata-ingestion/src/datahub/api/entities/assertion/volume_assertion.py
@@ -64,7 +64,7 @@ class RowCountChangeVolumeAssertion(BaseEntityAssertion):
                 rowCountChange=RowCountChange(
                     type=(
                         AssertionValueChangeType.ABSOLUTE
-                        if self.change_type == Literal["absolute"]
+                        if self.change_type == "absolute"
                         else AssertionValueChangeType.PERCENTAGE
                     ),
                     operator=self.operator.operator,