Fix memory issue.
