3 files changed, 7 insertions, 5 deletions

diff --git a/drivers/block/xen-blkfront.c b/drivers/block/xen-blkfront.c
index 9fd3ee203b1e..4276ab035e00 100644
--- a/drivers/block/xen-blkfront.c
+++ b/drivers/block/xen-blkfront.c
@@ -1452,6 +1452,9 @@ static int __init xlblk_init(void)
         if (!xen_domain())
                 return -ENODEV;
 
+        if (!xen_platform_pci_unplug)
+                return -ENODEV;
+
         if (register_blkdev(XENVBD_MAJOR, DEV_NAME)) {
                 printk(KERN_WARNING "xen_blk: can't get major %d with name %s\n",
                        XENVBD_MAJOR, DEV_NAME);
diff --git a/drivers/net/xen-netfront.c b/drivers/net/xen-netfront.c
index fa679057630f..8cc0914b9f92 100644
--- a/drivers/net/xen-netfront.c
+++ b/drivers/net/xen-netfront.c
@@ -47,6 +47,7 @@
 #include <xen/xenbus.h>
 #include <xen/events.h>
 #include <xen/page.h>
+#include <xen/platform_pci.h>
 #include <xen/grant_table.h>
 
 #include <xen/interface/io/netif.h>
@@ -1956,6 +1957,9 @@ static int __init netif_init(void)
         if (xen_initial_domain())
                 return 0;
 
+        if (!xen_platform_pci_unplug)
+                return -ENODEV;
+
         printk(KERN_INFO "Initialising Xen virtual ethernet driver.\n");
 
         return xenbus_register_frontend(&netfront_driver);
diff --git a/drivers/xen/platform-pci.c b/drivers/xen/platform-pci.c
index 319dd0a94d51..2389e581e23c 100644
--- a/drivers/xen/platform-pci.c
+++ b/drivers/xen/platform-pci.c
@@ -186,11 +186,6 @@ static struct pci_driver platform_driver = {
 
 static int __init platform_pci_module_init(void)
 {
-        /* no unplug has been done, IGNORE hasn't been specified: just
-         * return now */
-        if (!xen_platform_pci_unplug)
-                return -ENODEV;
-
         return pci_register_driver(&platform_driver);
 }