diff --git a/protos/feast/core/Transformation.proto b/protos/feast/core/Transformation.proto
index cde2833fa4..36f1e691fe 100644
--- a/protos/feast/core/Transformation.proto
+++ b/protos/feast/core/Transformation.proto
@@ -21,13 +21,12 @@ message UserDefinedFunctionV2 {
 
 // A feature transformation executed as a user-defined function
 message FeatureTransformationV2 {
-  // Note this Transformation starts at 5 for backwards compatibility
   oneof transformation {
     UserDefinedFunctionV2 user_defined_function = 1;
-    OnDemandSubstraitTransformationV2 on_demand_substrait_transformation = 2;
+    SubstraitTransformationV2 substrait_transformation = 2;
   }
 }
 
-message OnDemandSubstraitTransformationV2 {
+message SubstraitTransformationV2 {
   bytes substrait_plan = 1;
 }