Fix tests

This should allow the module to pass Travis, and get it onto the main Boxen org in GitHub.