fix the mem bug