if (!pwm)
                return -EINVAL;
 
-       if (!test_and_set_bit(PWMF_ENABLED, &pwm->flags)) {
+       if (!pwm_is_enabled(pwm)) {
                err = pwm->chip->ops->enable(pwm->chip, pwm);
-               if (err)
-                       clear_bit(PWMF_ENABLED, &pwm->flags);
+               if (!err)
+                       pwm->state.enabled = true;
        }
 
        return err;
  */
 void pwm_disable(struct pwm_device *pwm)
 {
-       if (pwm && test_and_clear_bit(PWMF_ENABLED, &pwm->flags))
+       if (!pwm)
+               return;
+
+       if (pwm_is_enabled(pwm)) {
                pwm->chip->ops->disable(pwm->chip, pwm);
+               pwm->state.enabled = false;
+       }
 }
 EXPORT_SYMBOL_GPL(pwm_disable);
 
 
 
 enum {
        PWMF_REQUESTED = 1 << 0,
-       PWMF_ENABLED = 1 << 1,
-       PWMF_EXPORTED = 1 << 2,
+       PWMF_EXPORTED = 1 << 1,
 };
 
 /*
  * @period: PWM period (in nanoseconds)
  * @duty_cycle: PWM duty cycle (in nanoseconds)
  * @polarity: PWM polarity
+ * @enabled: PWM enabled status
  */
 struct pwm_state {
        unsigned int period;
        unsigned int duty_cycle;
        enum pwm_polarity polarity;
+       bool enabled;
 };
 
 /**
 
 static inline bool pwm_is_enabled(const struct pwm_device *pwm)
 {
-       return test_bit(PWMF_ENABLED, &pwm->flags);
+       struct pwm_state state;
+
+       pwm_get_state(pwm, &state);
+
+       return state.enabled;
 }
 
 static inline void pwm_set_period(struct pwm_device *pwm, unsigned int period)