diff --git a/api.go b/api.go
index bbd8287..af0136f 100644
--- a/api.go
+++ b/api.go
@@ -1,7 +1,7 @@
 package dns_api
 
 import (
-	"crypto/ed25519"
+	"git.gammaspectra.live/givna.me/dns-api/ed25519"
 	"net/http"
 	"time"
 )
diff --git a/identity.go b/identity.go
index df6232d..8fa1dcc 100644
--- a/identity.go
+++ b/identity.go
@@ -2,10 +2,10 @@ package dns_api
 
 import (
 	"bytes"
-	"crypto/ed25519"
 	"encoding/hex"
 	"errors"
 	"fmt"
+	"git.gammaspectra.live/givna.me/dns-api/ed25519"
 	"net/url"
 	"sort"
 	"strconv"
@@ -57,8 +57,7 @@ func CreateSignatureMessage(method string, host string, requestUrl *url.URL, pri
 		return nil, err
 	}
 
-	publicKey := make([]byte, ed25519.PublicKeySize)
-	copy(publicKey, privateKey[ed25519.PublicKeySize:])
+	publicKey := privateKey.Public().(ed25519.PublicKey)
 
 	newRequestUrl.Query().Set(KeyPublicKey, hex.EncodeToString(publicKey))
 	newRequestUrl.Query().Set(KeyExpiration, strconv.FormatInt(expirationTime.UTC().Unix(), 10))