Fix bug with jquery