fix ui issue