diff --git a/bbswitch.c b/bbswitch.c
index 944fcf1..d23e024 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_D3hot);
+    pci_set_power_state(dis_dev, PCI_D3cold);
 
     if (bbswitch_acpi_off())
         pr_warn("The discrete card could not be disabled by a _DSM call\n");
@@ -436,7 +436,7 @@ static int __init bbswitch_init(void) {
                 dev_name(&pdev->dev), (char *)buf.pointer);
         } else {
             dis_dev = pdev;
-            pci_d3cold_disable(pdev);
+            pci_d3cold_enable(pdev);
             dis_handle = handle;
             pr_info("Found discrete VGA device %s: %s\n",
                 dev_name(&pdev->dev), (char *)buf.pointer);