Yeah this is ridiculous that this needs something like this.
Forgot the vmap.
Shut up fg.