[Fix] Save UI bugs