diff --git a/libraries/onos.robot b/libraries/onos.robot
index 72d7c8a..77f4091 100755
--- a/libraries/onos.robot
+++ b/libraries/onos.robot
@@ -399,11 +399,11 @@
     Should Be Equal As Integers    ${allocations}    ${count}
 
 Validate Subscriber DHCP Allocation
-    [Arguments]    ${ip}    ${port}    ${onu_port}
+    [Arguments]    ${ip}    ${port}    ${onu_port}   ${vlan}=''
     [Documentation]    Verifies that the specified subscriber is found in DHCP allocations
     ##TODO: Enhance the keyword to include DHCP allocated address is not 0.0.0.0
     ${allocations}=    Execute ONOS CLI Command    ${ip}    ${port}
-    ...    dhcpl2relay-allocations | grep DHCPACK | grep ${onu_port}
+    ...    dhcpl2relay-allocations | grep DHCPACK | grep ${onu_port} | grep ${vlan}
     Should Not Be Empty    ${allocations}    ONU port ${onu_port} not found in dhcpl2relay-allocations
 
 Device Is Available In ONOS