Fix UI formatting