fix wrap bug