Change the way we load the ui files. Just pass it to get_ui_class.