fix ui issue
fix ui issue