Fixing a dumb bug and adding support for more external use of vit