Minor fix
.end() required for correct context