Fix refresh UI