diff --git a/.travis.yml b/.travis.yml
index b686ec7..a3d8ce6 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -13,7 +13,7 @@ matrix:
     - dep ensure
     - go test -v $(go list ./...)
     - go vet $(go list ./...)
-    - diff <(goimports -d $(find . -type f -name '*.go' -not -path "./vendor/*")) <(printf "")
+    - diff <(goimports -local github.com/chihaya/chihaya -d $(find . -type f -name '*.go' -not -path "./vendor/*")) <(printf "")
     - (for d in $(go list ./...); do diff <(golint $d) <(printf "") || exit 1;  done)
     - go install github.com/chihaya/chihaya/cmd/chihaya
     - chihaya --config=example_config.yaml --debug&
@@ -29,7 +29,7 @@ matrix:
     script:
     - go test -v $(go list ./...)
     - go vet $(go list ./...)
-    - diff <(goimports -d $(find . -type f -name '*.go' -not -path "./vendor/*")) <(printf "")
+    - diff <(goimports -local github.com/chihaya/chihaya -d $(find . -type f -name '*.go' -not -path "./vendor/*")) <(printf "")
     - (for d in $(go list ./...); do diff <(golint $d) <(printf "") || exit 1;  done)
     - go install github.com/chihaya/chihaya/cmd/chihaya
     - chihaya --config=example_config.yaml --debug&
@@ -46,7 +46,7 @@ matrix:
     script:
     - go test -v $(go list ./...)
     - go vet $(go list ./...)
-    - diff <(goimports -d $(find . -type f -name '*.go' -not -path "./vendor/*")) <(printf "")
+    - diff <(goimports -local github.com/chihaya/chihaya -d $(find . -type f -name '*.go' -not -path "./vendor/*")) <(printf "")
     - (for d in $(go list ./...); do diff <(golint $d) <(printf "") || exit 1;  done)
     - go install github.com/chihaya/chihaya/cmd/chihaya
     - chihaya --config=example_config.yaml --debug&
diff --git a/cmd/chihaya/config.go b/cmd/chihaya/config.go
index 23dcc08..86daa7c 100644
--- a/cmd/chihaya/config.go
+++ b/cmd/chihaya/config.go
@@ -10,14 +10,18 @@ import (
 	"github.com/chihaya/chihaya/frontend/http"
 	"github.com/chihaya/chihaya/frontend/udp"
 	"github.com/chihaya/chihaya/middleware"
+)
 
-	// Imported to register as middleware drivers.
+import (
+	// Imports to register middleware drivers.
 	_ "github.com/chihaya/chihaya/middleware/clientapproval"
 	_ "github.com/chihaya/chihaya/middleware/jwt"
 	_ "github.com/chihaya/chihaya/middleware/torrentapproval"
 	_ "github.com/chihaya/chihaya/middleware/varinterval"
+)
 
-	// Imported to register as storage drivers.
+import (
+	// Imports to register storage drivers.
 	_ "github.com/chihaya/chihaya/storage/memory"
 	_ "github.com/chihaya/chihaya/storage/memorybysubnet"
 )
diff --git a/frontend/udp/connection_id_test.go b/frontend/udp/connection_id_test.go
index 044d7db..15f68eb 100644
--- a/frontend/udp/connection_id_test.go
+++ b/frontend/udp/connection_id_test.go
@@ -10,7 +10,6 @@ import (
 	"time"
 
 	"github.com/minio/sha256-simd"
-
 	"github.com/stretchr/testify/require"
 
 	"github.com/chihaya/chihaya/pkg/log"
diff --git a/middleware/varinterval/varinterval_test.go b/middleware/varinterval/varinterval_test.go
index 0ce8c07..d461eff 100644
--- a/middleware/varinterval/varinterval_test.go
+++ b/middleware/varinterval/varinterval_test.go
@@ -5,8 +5,9 @@ import (
 	"fmt"
 	"testing"
 
-	"github.com/chihaya/chihaya/bittorrent"
 	"github.com/stretchr/testify/require"
+
+	"github.com/chihaya/chihaya/bittorrent"
 )
 
 var configTests = []struct {
diff --git a/pkg/prometheus/server.go b/pkg/prometheus/server.go
index 1330a9f..3ce8903 100644
--- a/pkg/prometheus/server.go
+++ b/pkg/prometheus/server.go
@@ -6,9 +6,10 @@ import (
 	"context"
 	"net/http"
 
+	"github.com/prometheus/client_golang/prometheus"
+
 	"github.com/chihaya/chihaya/pkg/log"
 	"github.com/chihaya/chihaya/pkg/stop"
-	"github.com/prometheus/client_golang/prometheus"
 )
 
 // Server represents a standalone HTTP server for serving a Prometheus metrics
diff --git a/storage/memory/peer_store_test.go b/storage/memory/peer_store_test.go
index 1a2aee2..25b2ec4 100644
--- a/storage/memory/peer_store_test.go
+++ b/storage/memory/peer_store_test.go
@@ -2,7 +2,6 @@ package memory
 
 import (
 	"testing"
-
 	"time"
 
 	s "github.com/chihaya/chihaya/storage"