Update entire StateDocument in memory after updating in Mongo