1 files changed, 9 insertions, 8 deletions
diff --git a/manifests/wireguard_peer.pp b/manifests/wireguard_peer.pp
index 51df0d8..ea4c65a 100644
--- a/manifests/wireguard_peer.pp
+++ b/manifests/wireguard_peer.pp
@@ -1,19 +1,20 @@
 class profiles::wireguard_peer (
   Variant[String,Sensitive[String]] $private_key,
   Array[Hash] $peers,
+  Stdlib::IP::Address $network_address,
   String $ifname = 'wg0',
 ) {
   include ::profiles::wireguard
 
   networking::networkd_instance { $ifname:
-    type              => 'netdev',
-    content           => {
+    type    => 'netdev',
+    content => {
       'NetDev'        => {
         'Name'        => $ifname,
         'Kind'        => 'wireguard',
         'Description' => "WireGuard tunnel ${ifname}"
       },
-      'WireGuard'    => {
+      'WireGuard'     => {
         'PrivateKey' => $private_key,
       },
       'WireGuardPeer' => $peers,
@@ -21,13 +22,13 @@ class profiles::wireguard_peer (
   }
 
   networking::networkd_instance { "${ifname}-network":
-    type      => 'network',
-    content   => {
-      'Match' => {
+    type    => 'network',
+    content => {
+      'Match'   => {
         'Name' => $ifname,
       },
-      'Network'   => {
-        'Address' => '2001:9b1:eff:a600:22cf:30ff:fe45:629e/128',
+      'Network' => {
+        'Address' => $network_address,
       },
     }
   }