1 files changed, 3 insertions, 2 deletions
diff --git a/device/cookie.go b/device/cookie.go
index f134128..ec54f61 100644
--- a/device/cookie.go
+++ b/device/cookie.go
@@ -13,6 +13,7 @@ import (
 
 	"golang.org/x/crypto/blake2s"
 	"golang.org/x/crypto/chacha20poly1305"
+	"golang.zx2c4.com/wireguard/wgcfg"
 )
 
 type CookieChecker struct {
@@ -41,7 +42,7 @@ type CookieGenerator struct {
 	}
 }
 
-func (st *CookieChecker) Init(pk NoisePublicKey) {
+func (st *CookieChecker) Init(pk wgcfg.Key) {
 	st.Lock()
 	defer st.Unlock()
 
@@ -171,7 +172,7 @@ func (st *CookieChecker) CreateReply(
 	return reply, nil
 }
 
-func (st *CookieGenerator) Init(pk NoisePublicKey) {
+func (st *CookieGenerator) Init(pk wgcfg.Key) {
 	st.Lock()
 	defer st.Unlock()