Simplify Code

Since it's just a state-toogler, use the return value of the
method to set the values directly.

Signed-off-by: Tomaz Canabrava <tomaz.canabrava@intel.com>