diff --git a/kitserver/handler_test.go b/kitserver/handler_test.go
index 892fb32f41..3678c0acad 100644
--- a/kitserver/handler_test.go
+++ b/kitserver/handler_test.go
@@ -1,13 +1,13 @@
 package kitserver
 
 import (
-	"context"
 	"net/http/httptest"
 	"testing"
 
 	"github.com/gorilla/mux"
 	"github.com/kolide/kolide-ose/datastore"
 	"github.com/stretchr/testify/assert"
+	"golang.org/x/net/context"
 )
 
 func TestAPIRoutes(t *testing.T) {