diff --git a/bbswitch.c b/bbswitch.c
index c0c6713..e5b8c98 100644
--- a/bbswitch.c
+++ b/bbswitch.c
@@ -276,7 +276,7 @@ static void bbswitch_off(void) {
             ad->power.state = ACPI_STATE_D0;
         }
     } while (0);
-    pci_set_power_state(dis_dev, PCI_D3cold);
+    pci_set_power_state(dis_dev, PCI_D3hot);
 
     if (bbswitch_acpi_off())
         pr_warn("The discrete card could not be disabled by a _DSM call\n");
@@ -430,7 +430,7 @@ static int __init bbswitch_init(void) {
 
         acpi_get_name(handle, ACPI_FULL_PATHNAME, &buf);
 
-        if (pdev->vendor == PCI_VENDOR_ID_INTEL) {
+        if (pdev->vendor == PCI_VENDOR_ID_ATI) {
             igd_handle = handle;
             pr_info("Found integrated VGA device %s: %s\n",
                 dev_name(&pdev->dev), (char *)buf.pointer);