wed->wlan.base = devm_ioremap(dev->mt76.dev,
                                              pci_resource_start(pci_dev, 0),
                                              pci_resource_len(pci_dev, 0));
+               if (!wed->wlan.base)
+                       return -ENOMEM;
+
                wed->wlan.phy_base = pci_resource_start(pci_dev, 0);
                wed->wlan.wpdma_int = pci_resource_start(pci_dev, 0) +
                                      MT_INT_WED_SOURCE_CSR;
                wed->wlan.bus_type = MTK_WED_BUS_AXI;
                wed->wlan.base = devm_ioremap(dev->mt76.dev, res->start,
                                              resource_size(res));
+               if (!wed->wlan.base)
+                       return -ENOMEM;
+
                wed->wlan.phy_base = res->start;
                wed->wlan.wpdma_int = res->start + MT_INT_SOURCE_CSR;
                wed->wlan.wpdma_mask = res->start + MT_INT_MASK_CSR;