Solve UI problem