Fix mem bug