Fix updating context dict