1 files changed, 4 insertions, 10 deletions

diff --git a/arch/mips/pci/pci-lasat.c b/arch/mips/pci/pci-lasat.c
index e70ae3236e0b..a98e543a514a 100644
--- a/arch/mips/pci/pci-lasat.c
+++ b/arch/mips/pci/pci-lasat.c
@@ -10,7 +10,7 @@
 #include <linux/pci.h>
 #include <linux/types.h>
 
-#include <asm/bootinfo.h>
+#include <asm/lasat/lasat.h>
 
 #include <irq.h>
 
@@ -39,16 +39,10 @@ static int __init lasat_pci_setup(void)
 {
         printk(KERN_DEBUG "PCI: starting\n");
 
-        switch (mips_machtype) {
-        case MACH_LASAT_100:
-                lasat_pci_controller.pci_ops = &gt64xxx_pci0_ops;
-                break;
-        case MACH_LASAT_200:
+        if (IS_LASAT_200())
                 lasat_pci_controller.pci_ops = &nile4_pci_ops;
-                break;
-        default:
-                panic("pcibios_init: mips_machtype incorrect");
-        }
+        else
+                lasat_pci_controller.pci_ops = &gt64xxx_pci0_ops;
 
         register_pci_controller(&lasat_pci_controller);