fix jquery code