Corrected incorrect package on new UI class