}
 }
 
-static inline unsigned int
-ps2_test_one(struct ps2if *ps2if, unsigned int mask)
+static unsigned int __devinit ps2_test_one(struct ps2if *ps2if,
+                                          unsigned int mask)
 {
        unsigned int val;
 
  * Test the keyboard interface.  We basically check to make sure that
  * we can drive each line to the keyboard independently of each other.
  */
-static int __init ps2_test(struct ps2if *ps2if)
+static int __devinit ps2_test(struct ps2if *ps2if)
 {
        unsigned int stat;
        int ret = 0;
 /*
  * Remove one device from this driver.
  */
-static int ps2_remove(struct sa1111_dev *dev)
+static int __devexit ps2_remove(struct sa1111_dev *dev)
 {
        struct ps2if *ps2if = sa1111_get_drvdata(dev);
 
        },
        .devid          = SA1111_DEVID_PS2,
        .probe          = ps2_probe,
-       .remove         = ps2_remove,
+       .remove         = __devexit_p(ps2_remove),
 };
 
 static int __init ps2_init(void)