[LMB]: Fix lmb allocation regression.