Fix JQuery script issue