Change document.getElementById by jQuery selector.