Minor update.
Handle memory leak.