Fix JQuery bug