Fix document.ready