fix mem bug