Minor jquery fix