Improved jQuery UI compatibility