Fix possible double-free issue