fix enable/disable flow