diff --git a/onnxruntime/core/providers/dml/DmlExecutionProvider/src/ReadbackHeap.cpp b/onnxruntime/core/providers/dml/DmlExecutionProvider/src/ReadbackHeap.cpp
index a3517e12e1..a02f0e49c7 100644
--- a/onnxruntime/core/providers/dml/DmlExecutionProvider/src/ReadbackHeap.cpp
+++ b/onnxruntime/core/providers/dml/DmlExecutionProvider/src/ReadbackHeap.cpp
@@ -63,7 +63,7 @@ namespace Dml
             m_readbackHeap = CreateReadbackHeap(m_device.Get(), m_capacity);
         }
 
-        assert(gsl::narrow_cast<ptrdiff_t>(m_readbackHeap->GetDesc().Width) >= totalSize);
+        assert(gsl::narrow_cast<ptrdiff_t>(m_readbackHeap->GetDesc().Width) >= size);
     }
 
     void ReadbackHeap::ReadbackFromGpu(