Solve ui issue on production