Small change to JQuery code to fix a bug.