#include <linux/device.h>
 #include <linux/mutex.h>
 #include <linux/i2c.h>
-#include <linux/gpio.h>
+#include <linux/gpio/driver.h>
 #include <linux/of.h>
 #include <linux/of_gpio.h>
 #include <linux/slab.h>
 
 static void mcu_gpio_set(struct gpio_chip *gc, unsigned int gpio, int val)
 {
-       struct mcu *mcu = container_of(gc, struct mcu, gc);
+       struct mcu *mcu = gpiochip_get_data(gc);
        u8 bit = 1 << (4 + gpio);
 
        mutex_lock(&mcu->lock);
        gc->direction_output = mcu_gpio_dir_out;
        gc->of_node = np;
 
-       return gpiochip_add(gc);
+       return gpiochip_add_data(gc, mcu);
 }
 
 static int mcu_gpiochip_remove(struct mcu *mcu)