bug fixed, UI back