Fix balloon update issue