* Copyright (C) 2009  Phil Sutter <n0-1@freewrt.org>
  */
 
-#include <linux/input-polldev.h>
+#include <linux/input.h>
 #include <linux/module.h>
 #include <linux/platform_device.h>
 #include <linux/gpio.h>
        return !val;
 }
 
-static void rb532_button_poll(struct input_polled_dev *poll_dev)
+static void rb532_button_poll(struct input_dev *input)
 {
-       input_report_key(poll_dev->input, RB532_BTN_KSYM,
-                        rb532_button_pressed());
-       input_sync(poll_dev->input);
+       input_report_key(input, RB532_BTN_KSYM, rb532_button_pressed());
+       input_sync(input);
 }
 
 static int rb532_button_probe(struct platform_device *pdev)
 {
-       struct input_polled_dev *poll_dev;
+       struct input_dev *input;
        int error;
 
-       poll_dev = devm_input_allocate_polled_device(&pdev->dev);
-       if (!poll_dev)
+       input = devm_input_allocate_device(&pdev->dev);
+       if (!input)
                return -ENOMEM;
 
-       poll_dev->poll = rb532_button_poll;
-       poll_dev->poll_interval = RB532_BTN_RATE;
+       input->name = "rb532 button";
+       input->phys = "rb532/button0";
+       input->id.bustype = BUS_HOST;
 
-       poll_dev->input->name = "rb532 button";
-       poll_dev->input->phys = "rb532/button0";
-       poll_dev->input->id.bustype = BUS_HOST;
+       input_set_capability(input, EV_KEY, RB532_BTN_KSYM);
 
-       input_set_capability(poll_dev->input, EV_KEY, RB532_BTN_KSYM);
+       error = input_setup_polling(input, rb532_button_poll);
+       if (error)
+               return error;
+
+       input_set_poll_interval(input, RB532_BTN_RATE);
 
-       error = input_register_polled_device(poll_dev);
+       error = input_register_device(input);
        if (error)
                return error;