fixed vmap tests.