fixed the ui issue