Function state is not set correct

[am1337]

When switching to loco control mode already enabled functions are not marked as enabled.
But when a state change is sent by server while in loco control mode it is updated on oled button.