Fix problem with Jquery UI