Refactor test for JsonDoc.modify