fix TX memory assign