Refactored the code based on `vagrant-omnibus`

This seems to fix the errors that are occurring with Vagrant 1.4.*