Fix UI refesh