fix mem bugs