Fixed problem with jQuery UI