fix the error in vmap about C-S