1 files changed, 2 insertions, 2 deletions
diff --git a/dhcpv4/ztpv4/ztp_test.go b/dhcpv4/ztpv4/ztp_test.go
index 9950bf5..680f15a 100644
--- a/dhcpv4/ztpv4/ztp_test.go
+++ b/dhcpv4/ztpv4/ztp_test.go
@@ -54,13 +54,13 @@ func TestParseV4VendorClass(t *testing.T) {
 			}
 
 			if tc.vc != "" {
-				packet.AddOption(&dhcpv4.OptClassIdentifier{
+				packet.UpdateOption(&dhcpv4.OptClassIdentifier{
 					Identifier: tc.vc,
 				})
 			}
 
 			if tc.hostname != "" {
-				packet.AddOption(&dhcpv4.OptHostName{
+				packet.UpdateOption(&dhcpv4.OptHostName{
 					HostName: tc.hostname,
 				})
 			}