diff --git a/drivers/nvme/host/pci.c b/drivers/nvme/host/pci.c
index 8a15aa50b8e066d026bc4083ff7e75c94b7b8bf1..58dbe684007bdcc7e0b73408519418fd469ab389 100644
--- a/drivers/nvme/host/pci.c
+++ b/drivers/nvme/host/pci.c
@@ -1787,7 +1787,7 @@ static int __nvme_alloc_host_mem(struct nvme_dev *dev, u64 preferred,
 	if (!bufs)
 		goto out_free_descs;
 
-	for (size = 0; size < preferred; size += len) {
+	for (size = 0; size < preferred && i < max_entries; size += len) {
 		dma_addr_t dma_addr;
 
 		len = min_t(u64, chunk_size, preferred - size);