fix funciton wrap bug