Fix code potential errors.

[mirror_edk2.git] / MdeModulePkg / Bus / Ata / AtaBusDxe / AtaBus.c

diff --git a/MdeModulePkg/Bus/Ata/AtaBusDxe/AtaBus.c b/MdeModulePkg/Bus/Ata/AtaBusDxe/AtaBus.c
index 625ba1b5b18b4527e6fb7ba24765b8541b8fe076..51687fe0d322ab2e066d898682fcd44009bd09d0 100644 (file)
--- a/MdeModulePkg/Bus/Ata/AtaBusDxe/AtaBus.c
+++ b/MdeModulePkg/Bus/Ata/AtaBusDxe/AtaBus.c
@@ -464,6 +464,7 @@ UnregisterAtaDevice (
   if (BlockIo != NULL) {\r
     AtaDevice = ATA_DEVICE_FROM_BLOCK_IO (BlockIo);\r
   } else {\r
+    ASSERT (BlockIo2 != NULL);\r
     AtaDevice = ATA_DEVICE_FROM_BLOCK_IO2 (BlockIo2);\r
   }\r
 \r