Fix mem bugs