correct the stan code in vignette