#include <linux/regulator/mt6358-regulator.h>
 #include <linux/regulator/of_regulator.h>
 
-#define MT6358_BUCK_MODE_AUTO  0
-#define MT6358_BUCK_MODE_FORCE_PWM     1
+#include <dt-bindings/regulator/mediatek,mt6397-regulator.h>
 
 /*
  * MT6358 regulators' information
 
 static unsigned int mt6358_map_mode(unsigned int mode)
 {
-       return mode == MT6358_BUCK_MODE_AUTO ?
+       return mode == MT6397_BUCK_MODE_AUTO ?
                REGULATOR_MODE_NORMAL : REGULATOR_MODE_FAST;
 }
 
 
        switch (mode) {
        case REGULATOR_MODE_FAST:
-               val = MT6358_BUCK_MODE_FORCE_PWM;
+               val = MT6397_BUCK_MODE_FORCE_PWM;
                break;
        case REGULATOR_MODE_NORMAL:
-               val = MT6358_BUCK_MODE_AUTO;
+               val = MT6397_BUCK_MODE_AUTO;
                break;
        default:
                return -EINVAL;
        }
 
        switch ((regval & info->modeset_mask) >> (ffs(info->modeset_mask) - 1)) {
-       case MT6358_BUCK_MODE_AUTO:
+       case MT6397_BUCK_MODE_AUTO:
                return REGULATOR_MODE_NORMAL;
-       case MT6358_BUCK_MODE_FORCE_PWM:
+       case MT6397_BUCK_MODE_FORCE_PWM:
                return REGULATOR_MODE_FAST;
        default:
                return -EINVAL;