fix allocation issue just introduced