diff --git a/tests/arp/test_wr_arp.py b/tests/arp/test_wr_arp.py
index bfae665c3a4..a4dcb174b1a 100644
--- a/tests/arp/test_wr_arp.py
+++ b/tests/arp/test_wr_arp.py
@@ -200,7 +200,8 @@ def prepareSshKeys(self, duthost, ptfhost):
             Returns:
                 None
         '''
-        invetory = duthost.host.options['inventory'].split('/')[-1]
+        hostVars = duthost.host.options['variable_manager']._hostvars[duthost.hostname]
+        invetory = hostVars['inventory_file'].split('/')[-1]
         secrets = duthost.host.options['variable_manager']._hostvars[duthost.hostname]['secret_group_vars']
 
         prepareTestbedSshKeys(duthost, ptfhost, secrets[invetory]['sonicadmin_user'])