Fix UI display