Fix document ready call