diff --git a/src/test/java/org/opencord/kafka/integrations/DhcpL2RelayKafkaIntegrationTest.java b/src/test/java/org/opencord/kafka/integrations/DhcpL2RelayKafkaIntegrationTest.java
index fe45d40..3a4ed77 100644
--- a/src/test/java/org/opencord/kafka/integrations/DhcpL2RelayKafkaIntegrationTest.java
+++ b/src/test/java/org/opencord/kafka/integrations/DhcpL2RelayKafkaIntegrationTest.java
@@ -68,7 +68,7 @@
         DhcpAllocationInfo allocationInfo = new DhcpAllocationInfo(
                 OLT_CONNECT_POINT,
                 DHCP.MsgType.DHCPREQUEST, ONU_SERIAL, OLT_MAC,
-                LOCAL_IP);
+                LOCAL_IP, SUBSCRIBER_ID);
         DhcpL2RelayEvent event = new DhcpL2RelayEvent(DhcpL2RelayEvent.Type.STATS_UPDATE,
                 allocationInfo, OLT_CONNECT_POINT, entryCounter, null);
         dhcpL2RelayListener.event(event);
@@ -86,7 +86,7 @@
         DhcpAllocationInfo allocationInfo = new DhcpAllocationInfo(
                 OLT_CONNECT_POINT,
                 DHCP.MsgType.DHCPREQUEST, ONU_SERIAL, OLT_MAC,
-                LOCAL_IP);
+                LOCAL_IP, SUBSCRIBER_ID);
         DhcpL2RelayEvent event = new DhcpL2RelayEvent(DhcpL2RelayEvent.Type.STATS_UPDATE,
                 allocationInfo, OLT_CONNECT_POINT, entryCounter, ONU_SERIAL);
         dhcpL2RelayListener.event(event);
@@ -104,7 +104,7 @@
         DhcpAllocationInfo allocationInfo = new DhcpAllocationInfo(
                 new ConnectPoint(DEVICE_ID_1, PORT.number()),
                 DHCP.MsgType.DHCPREQUEST, ONU_SERIAL, OLT_MAC,
-                LOCAL_IP);
+                LOCAL_IP, SUBSCRIBER_ID);
         DhcpL2RelayEvent event = new DhcpL2RelayEvent(DhcpL2RelayEvent.Type.UPDATED,
                 allocationInfo, cp, entryCounter, null);
         dhcpL2RelayListener.event(event);
@@ -161,7 +161,7 @@
         }
 
         @Override
-        public boolean removeAllocationByConnectPoint(ConnectPoint connectPoint) {
+        public boolean removeAllocationsByConnectPoint(ConnectPoint connectPoint) {
             return false;
         }
     }