Small change to prevent memory leaks