1 files changed, 4 insertions, 2 deletions
diff --git a/src/main/java/com/lumaserv/bgp/protocol/attribute/NextHopAttribute.java b/src/main/java/com/lumaserv/bgp/protocol/attribute/NextHopAttribute.java
index 2146364..c579295 100644
--- a/src/main/java/com/lumaserv/bgp/protocol/attribute/NextHopAttribute.java
+++ b/src/main/java/com/lumaserv/bgp/protocol/attribute/NextHopAttribute.java
@@ -6,6 +6,8 @@ import java.net.UnknownHostException;
 import lombok.Getter;
 import lombok.Setter;
 
+import java.nio.ByteBuffer;
+
 @Getter
 @Setter
 public class NextHopAttribute implements PathAttribute {
@@ -27,8 +29,8 @@ public class NextHopAttribute implements PathAttribute {
         return 3;
     }
 
-    public byte[] build() {
-        return address;
+    public void build(ByteBuffer b) {
+        b.put(address);
     }
 
     public String toString() {