diff --git a/sdk/python/feast/sort_key.py b/sdk/python/feast/sort_key.py
index 9df447f154..8a5a815ca6 100644
--- a/sdk/python/feast/sort_key.py
+++ b/sdk/python/feast/sort_key.py
@@ -33,7 +33,7 @@ class SortKey:
 
     name: str
     value_type: ValueType
-    default_sort_order: int
+    default_sort_order: SortOrder.Enum.ValueType
     tags: Dict[str, str]
     description: str
 
@@ -41,7 +41,7 @@ def __init__(
         self,
         name: str,
         value_type: ValueType,
-        default_sort_order: int = int(SortOrder.Enum.ASC),
+        default_sort_order: SortOrder.Enum.ValueType = SortOrder.ASC,
         tags: Optional[Dict[str, str]] = None,
         description: str = "",
     ):
@@ -79,7 +79,7 @@ def from_proto(cls, proto: SortKeyProto) -> "SortKey":
         return cls(
             name=proto.name,
             value_type=ValueType(proto.value_type),
-            default_sort_order=int(proto.default_sort_order),
+            default_sort_order=proto.default_sort_order,
             tags=dict(proto.tags),
             description=proto.description,
-        )
\ No newline at end of file
+        )
diff --git a/sdk/python/feast/sorted_feature_view.py b/sdk/python/feast/sorted_feature_view.py
index 6604be8ca2..5ca413888f 100644
--- a/sdk/python/feast/sorted_feature_view.py
+++ b/sdk/python/feast/sorted_feature_view.py
@@ -213,4 +213,4 @@ def from_proto(cls, sfv_proto):
                 )
             )
 
-        return sorted_feature_view
\ No newline at end of file
+        return sorted_feature_view