diff --git a/superset-frontend/src/pages/DvtNewTrainedTable/index.tsx b/superset-frontend/src/pages/DvtNewTrainedTable/index.tsx
index 5a44c1c12cbcb..3f997741dcef1 100644
--- a/superset-frontend/src/pages/DvtNewTrainedTable/index.tsx
+++ b/superset-frontend/src/pages/DvtNewTrainedTable/index.tsx
@@ -73,7 +73,7 @@ function DvtNewTainedTable() {
             target_column_name:
               newTainedTableAddSelector.targetColumnName?.value,
             time_column_name: newTainedTableAddSelector.timeColumnName?.value,
-            predictionNumber: newTainedTableAddSelector.predictionNumber
+            prediction_number: newTainedTableAddSelector.predictionNumber
               ? newTainedTableAddSelector.predictionNumber
               : undefined,
           }