diff --git a/velox/substrait/SubstraitToVeloxPlan.cpp b/velox/substrait/SubstraitToVeloxPlan.cpp
index dd39f78c0e89..6694203f41f5 100644
--- a/velox/substrait/SubstraitToVeloxPlan.cpp
+++ b/velox/substrait/SubstraitToVeloxPlan.cpp
@@ -457,7 +457,7 @@ std::shared_ptr<const core::PlanNode> SubstraitVeloxPlanConverter::toVeloxPlan(
       splitInfo->starts.emplace_back(file.start());
       splitInfo->lengths.emplace_back(file.length());
       auto format = file.format();
-      if (format == 0) {
+      if (format == 2 || format == 3) {
         splitInfo->format = dwio::common::FileFormat::ORC;
       } else if (format == 1) {
         splitInfo->format = dwio::common::FileFormat::PARQUET;
diff --git a/velox/substrait/proto/substrait/algebra.proto b/velox/substrait/proto/substrait/algebra.proto
index 48ca9d050ff9..1d3070912d6c 100644
--- a/velox/substrait/proto/substrait/algebra.proto
+++ b/velox/substrait/proto/substrait/algebra.proto
@@ -102,6 +102,8 @@ message ReadRel {
       enum FileFormat {
         FILE_FORMAT_UNSPECIFIED = 0;
         FILE_FORMAT_PARQUET = 1;
+        FILE_FORMAT_ORC = 2;
+        FILE_FORMAT_DWRF = 3;
       }
     }
   }