1 files changed, 3 insertions, 9 deletions
diff --git a/internal/wireguard/wireguard.go b/internal/wireguard/wireguard.go
index db20067..bb26b69 100644
--- a/internal/wireguard/wireguard.go
+++ b/internal/wireguard/wireguard.go
@@ -3,6 +3,8 @@ package wireguard
 import (
 	"fmt"
 	"regexp"
+
+	"github.com/jwijenbergh/eduvpn-common/internal/types"
 	"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
 )
 
@@ -10,7 +12,7 @@ func GenerateKey() (wgtypes.Key, error) {
 	key, keyErr := wgtypes.GeneratePrivateKey()
 
 	if keyErr != nil {
-		return key, &WireguardGenerateKeyError{Err: keyErr}
+		return key, &types.WrappedErrorMessage{Message: "failed generating WireGuard key", Err: keyErr}
 	}
 	return key, nil
 }
@@ -28,11 +30,3 @@ func ConfigAddKey(config string, key wgtypes.Key) string {
 	to_replace := fmt.Sprintf("%s\nPrivateKey = %s", interface_section, key.String())
 	return interface_re.ReplaceAllString(config, to_replace)
 }
-
-type WireguardGenerateKeyError struct {
-	Err error
-}
-
-func (e *WireguardGenerateKeyError) Error() string {
-	return fmt.Sprintf("failed generating Wireguard key with error: %v", e.Err)
-}