diff --git a/onnxruntime/core/providers/dml/OperatorAuthorHelper/OperatorHelper.cpp b/onnxruntime/core/providers/dml/OperatorAuthorHelper/OperatorHelper.cpp
index 78c045a4e1..07de6d1988 100644
--- a/onnxruntime/core/providers/dml/OperatorAuthorHelper/OperatorHelper.cpp
+++ b/onnxruntime/core/providers/dml/OperatorAuthorHelper/OperatorHelper.cpp
@@ -971,7 +971,7 @@ namespace OperatorHelper
     std::vector<EdgeShapes> ConcatHelper::GetOutputShapes(const MLShapeInferenceContext& shapeInfo) const
     {
         auto outputShape = shapeInfo.GetInputTensorShape(0);
-        ML_CHECK_VALID_ARGUMENT(outputShape.size() <= NchwDimensionCount);
+        ML_CHECK_VALID_ARGUMENT(outputShape.size() <= NcdhwDimensionCount);
 
         uint32_t inputCount = shapeInfo.GetInputCount();