diff --git a/.buffalo.dev.yml b/.buffalo.dev.yml
new file mode 100644
index 0000000..a091fe8
--- /dev/null
+++ b/.buffalo.dev.yml
@@ -0,0 +1,21 @@
+app_root: .
+ignored_folders:
+- vendor
+- log
+- logs
+- assets
+- public
+- grifts
+- tmp
+- bin
+- node_modules
+- .sass-cache
+included_extensions:
+- .go
+- .env
+build_path: tmp
+build_delay: 200ns
+binary_name: swagip-build
+command_flags: []
+enable_colors: true
+log_name: buffalo
diff --git a/.travis.yml b/.travis.yml
index c41f998..0e84afe 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -17,3 +17,29 @@ install:
   - go get $(go list ./... | grep -v /vendor/)
 
 script: buffalo test
+
+
+
+language: go
+
+go:
+  - "1.11.x"
+
+env:
+  - GO_ENV=test
+
+before_script:
+  - mkdir -p $TRAVIS_BUILD_DIR/public/assets
+
+go_import_path: github.com/StackFocus/SwagIP-go
+
+install:
+  - go get github.com/gobuffalo/buffalo/buffalo
+{{- if .opts.App.WithDep }}
+  - go get github.com/golang/dep/cmd/dep
+  - dep ensure
+{{- else }}
+  - go get $(go list ./... | grep -v /vendor/)
+{{- end }}
+
+script: buffalo test
\ No newline at end of file
diff --git a/actions/home_test.go b/actions/home_test.go
index fc664cc..b2dde7f 100644
--- a/actions/home_test.go
+++ b/actions/home_test.go
@@ -1,8 +1,10 @@
 package actions
 
 func (as *ActionSuite) Test_CLIRootHandler() {
-	as.Willie.Headers["User-Agent"] = "curl"
-	res := as.JSON("/").Get()
+	req := as.JSON("/")
+	req.Headers["User-Agent"] = "curl"
+
+	res := req.Get()
 	as.Equal(200, res.Code)
 	as.Equal("", res.Body.String())
 }
@@ -14,8 +16,9 @@ func (as *ActionSuite) Test_RootHandler() {
 }
 
 func (as *ActionSuite) Test_AllHeadersHandler() {
-	as.Willie.Headers["User-Agent"] = "curl"
-	res := as.HTML("/all").Get()
+	req := as.HTML("/all")
+	req.Headers["User-Agent"] = "curl"
+	res := req.Get()
 	as.Equal(200, res.Code)
 	as.Contains(res.Body.String(), "user-agent: curl")
 }