Fix UI updating when control is reset