4 files changed, 10 insertions, 6 deletions

diff --git a/arch/powerpc/platforms/embedded6xx/gamecube.c b/arch/powerpc/platforms/embedded6xx/gamecube.c
index bd4ba5d7d568..fe0ed6ee285e 100644
--- a/arch/powerpc/platforms/embedded6xx/gamecube.c
+++ b/arch/powerpc/platforms/embedded6xx/gamecube.c
@@ -67,6 +67,8 @@ static int __init gamecube_probe(void)
         if (!of_flat_dt_is_compatible(dt_root, "nintendo,gamecube"))
                 return 0;
 
+        pm_power_off = gamecube_power_off;
+
         return 1;
 }
 
@@ -80,7 +82,6 @@ define_machine(gamecube) {
         .probe                  = gamecube_probe,
         .init_early             = gamecube_init_early,
         .restart                = gamecube_restart,
-        .power_off              = gamecube_power_off,
         .halt                   = gamecube_halt,
         .init_IRQ               = flipper_pic_probe,
         .get_irq                = flipper_pic_get_irq,
diff --git a/arch/powerpc/platforms/embedded6xx/linkstation.c b/arch/powerpc/platforms/embedded6xx/linkstation.c
index 168e1d80b2e5..540eeb58d3f0 100644
--- a/arch/powerpc/platforms/embedded6xx/linkstation.c
+++ b/arch/powerpc/platforms/embedded6xx/linkstation.c
@@ -147,6 +147,9 @@ static int __init linkstation_probe(void)
 
         if (!of_flat_dt_is_compatible(root, "linkstation"))
                 return 0;
+
+        pm_power_off = linkstation_power_off;
+
         return 1;
 }
 
@@ -158,7 +161,6 @@ define_machine(linkstation){
         .show_cpuinfo           = linkstation_show_cpuinfo,
         .get_irq                = mpic_get_irq,
         .restart                = linkstation_restart,
-        .power_off              = linkstation_power_off,
         .halt                   = linkstation_halt,
         .calibrate_decr         = generic_calibrate_decr,
 };
diff --git a/arch/powerpc/platforms/embedded6xx/usbgecko_udbg.c b/arch/powerpc/platforms/embedded6xx/usbgecko_udbg.c
index 20a8ed91962e..7feb325b636b 100644
--- a/arch/powerpc/platforms/embedded6xx/usbgecko_udbg.c
+++ b/arch/powerpc/platforms/embedded6xx/usbgecko_udbg.c
@@ -247,7 +247,7 @@ void __init ug_udbg_init(void)
         np = of_find_compatible_node(NULL, NULL, "nintendo,flipper-exi");
         if (!np) {
                 udbg_printf("%s: EXI node not found\n", __func__);
-                goto done;
+                goto out;
         }
 
         exi_io_base = ug_udbg_setup_exi_io_base(np);
@@ -267,8 +267,8 @@ void __init ug_udbg_init(void)
         }
 
 done:
-        if (np)
-                of_node_put(np);
+        of_node_put(np);
+out:
         return;
 }
 
diff --git a/arch/powerpc/platforms/embedded6xx/wii.c b/arch/powerpc/platforms/embedded6xx/wii.c
index 388e29bab8f6..352592d3e44e 100644
--- a/arch/powerpc/platforms/embedded6xx/wii.c
+++ b/arch/powerpc/platforms/embedded6xx/wii.c
@@ -211,6 +211,8 @@ static int __init wii_probe(void)
         if (!of_flat_dt_is_compatible(dt_root, "nintendo,wii"))
                 return 0;
 
+        pm_power_off = wii_power_off;
+
         return 1;
 }
 
@@ -226,7 +228,6 @@ define_machine(wii) {
         .init_early             = wii_init_early,
         .setup_arch             = wii_setup_arch,
         .restart                = wii_restart,
-        .power_off              = wii_power_off,
         .halt                   = wii_halt,
         .init_IRQ               = wii_pic_probe,
         .get_irq                = flipper_pic_get_irq,