summaryrefslogtreecommitdiff
path: root/storage/class/disk/src/diskwmi.c
diff options
context:
space:
mode:
Diffstat (limited to 'storage/class/disk/src/diskwmi.c')
-rw-r--r--storage/class/disk/src/diskwmi.c4
1 files changed, 2 insertions, 2 deletions
diff --git a/storage/class/disk/src/diskwmi.c b/storage/class/disk/src/diskwmi.c
index 7503e5a1..de2d7eee 100644
--- a/storage/class/disk/src/diskwmi.c
+++ b/storage/class/disk/src/diskwmi.c
@@ -1739,7 +1739,7 @@ DiskInfoExceptionComplete(
//
// Reset byte count of transfer in SRB Extension.
//
- srbEx->DataTransferLength = Irp->MdlAddress->ByteCount;
+ srbEx->DataTransferLength = MmGetMdlByteCount(Irp->MdlAddress);
//
// Zero SRB statuses.
@@ -1768,7 +1768,7 @@ DiskInfoExceptionComplete(
//
// Reset byte count of transfer in SRB Extension.
//
- srb->DataTransferLength = Irp->MdlAddress->ByteCount;
+ srb->DataTransferLength = MmGetMdlByteCount(Irp->MdlAddress);
//
// Zero SRB statuses.