Update to json test code