diff --git a/toolkit/server/middleware.go b/toolkit/server/middleware.go
index a9f2fd1..a956d2d 100644
--- a/toolkit/server/middleware.go
+++ b/toolkit/server/middleware.go
@@ -6,8 +6,8 @@ import (
 	"net/http"
 	"strings"
 
-	"github.com/ucan-wg/go-ucan/delegation"
 	"github.com/ucan-wg/go-ucan/did"
+	"github.com/ucan-wg/go-ucan/token/delegation"
 )
 
 type Middleware func(http.Handler) http.Handler