fix association class ui