Fix Bug UI