Fix bug with checkbox