Fix JS with document ready