fix UI setup