return 0;
 }
 
-static int unittest_remove(struct platform_device *pdev)
+static void unittest_remove(struct platform_device *pdev)
 {
        struct device *dev = &pdev->dev;
        struct device_node *np = dev->of_node;
 
        dev_dbg(dev, "%s for node @%pOF\n", __func__, np);
-       return 0;
 }
 
 static const struct of_device_id unittest_match[] = {
 
 static struct platform_driver unittest_driver = {
        .probe                  = unittest_probe,
-       .remove                 = unittest_remove,
+       .remove_new             = unittest_remove,
        .driver = {
                .name           = "unittest",
                .of_match_table = of_match_ptr(unittest_match),
        return ret;
 }
 
-static int unittest_gpio_remove(struct platform_device *pdev)
+static void unittest_gpio_remove(struct platform_device *pdev)
 {
        struct unittest_gpio_dev *devptr = platform_get_drvdata(pdev);
        struct device *dev = &pdev->dev;
                gpiochip_remove(&devptr->chip);
 
        kfree(devptr);
-
-       return 0;
 }
 
 static const struct of_device_id unittest_gpio_id[] = {
 
 static struct platform_driver unittest_gpio_driver = {
        .probe  = unittest_gpio_probe,
-       .remove = unittest_gpio_remove,
+       .remove_new = unittest_gpio_remove,
        .driver = {
                .name           = "unittest-gpio",
                .of_match_table = of_match_ptr(unittest_gpio_id),
        return 0;
 }
 
-static int unittest_i2c_bus_remove(struct platform_device *pdev)
+static void unittest_i2c_bus_remove(struct platform_device *pdev)
 {
        struct device *dev = &pdev->dev;
        struct device_node *np = dev->of_node;
 
        dev_dbg(dev, "%s for node @%pOF\n", __func__, np);
        i2c_del_adapter(&std->adap);
-
-       return 0;
 }
 
 static const struct of_device_id unittest_i2c_bus_match[] = {
 
 static struct platform_driver unittest_i2c_bus_driver = {
        .probe                  = unittest_i2c_bus_probe,
-       .remove                 = unittest_i2c_bus_remove,
+       .remove_new             = unittest_i2c_bus_remove,
        .driver = {
                .name           = "unittest-i2c-bus",
                .of_match_table = of_match_ptr(unittest_i2c_bus_match),