Fixed a mem bug