fix the watermark

fix regression (memory problem) w.r.t. pre4