Ui elements subclass correct class

fixed the subclass thing. @hunterboerner