Update tests for json generator