Fix JSON generation