diff --git a/internal/pkg/application/dhcprelay_test.go b/internal/pkg/application/dhcprelay_test.go
index d27ee8e..0b3989f 100644
--- a/internal/pkg/application/dhcprelay_test.go
+++ b/internal/pkg/application/dhcprelay_test.go
@@ -1018,22 +1018,23 @@
 				}
 			case "ProcessUDP6Packet_DHCPv6MsgTypeAdvertise":
 				dhcpv6.MsgType = layers.DHCPv6MsgTypeAdvertise
-				pkt.EXPECT().Layer(layers.LayerTypeDHCPv6).Return(dhcpv6).Times(2)
+				pkt.EXPECT().Layer(layers.LayerTypeDHCPv6).Return(dhcpv6).AnyTimes()
 				if got := va.ProcessUDP6Packet(tt.args.cntx, tt.args.device, tt.args.port, tt.args.pkt); !reflect.DeepEqual(got, tt.want) {
 					t.Errorf("VoltApplication.ProcessUDP6Packet() = %v, want %v", got, tt.want)
 				}
 			case "ProcessUDP6Packet_DHCPv6MsgTypeRelayForward":
 				dhcpv6.MsgType = layers.DHCPv6MsgTypeRelayForward
-				pkt.EXPECT().Layer(layers.LayerTypeDHCPv6).Return(dhcpv6).Times(2)
+				pkt.EXPECT().Layer(layers.LayerTypeDHCPv6).Return(dhcpv6).AnyTimes()
 				if got := va.ProcessUDP6Packet(tt.args.cntx, tt.args.device, tt.args.port, tt.args.pkt); !reflect.DeepEqual(got, tt.want) {
 					t.Errorf("VoltApplication.ProcessUDP6Packet() = %v, want %v", got, tt.want)
 				}
 			case "ProcessUDP6Packet_DHCPv6MsgTypeRelayReply":
 				dhcpv6.MsgType = layers.DHCPv6MsgTypeRelayReply
-				pkt.EXPECT().Data().Times(1)
-				pkt.EXPECT().Layer(layers.LayerTypeEthernet).Return(eth).Times(1)
-				pkt.EXPECT().Layer(layers.LayerTypeIPv6).Return(ipv6).Times(1)
-				pkt.EXPECT().Layer(layers.LayerTypeUDP).Return(uup).Times(1)
+				pkt.EXPECT().Layer(layers.LayerTypeDHCPv6).Return(dhcpv6).AnyTimes()
+				pkt.EXPECT().Data().AnyTimes()
+				pkt.EXPECT().Layer(layers.LayerTypeEthernet).Return(eth).AnyTimes()
+				pkt.EXPECT().Layer(layers.LayerTypeIPv6).Return(ipv6).AnyTimes()
+				pkt.EXPECT().Layer(layers.LayerTypeUDP).Return(uup).AnyTimes()
 				if got := va.ProcessUDP6Packet(tt.args.cntx, tt.args.device, tt.args.port, tt.args.pkt); !reflect.DeepEqual(got, tt.want) {
 					t.Errorf("VoltApplication.ProcessUDP6Packet() = %v, want %v", got, tt.want)
 				}