Fixed mem bug