Fix memory leak in write_riemann