diff --git a/tests/platform_tests/test_memory_exhaustion.py b/tests/platform_tests/test_memory_exhaustion.py
index c39299b89c1..5ea66320124 100644
--- a/tests/platform_tests/test_memory_exhaustion.py
+++ b/tests/platform_tests/test_memory_exhaustion.py
@@ -49,7 +49,7 @@ def tearDown(self, duthosts, enum_rand_one_per_hwsku_hostname,
                           'Recover {} by PDU reboot failed'.format(hostname))
             # Wait until all critical processes are healthy.
             wait_critical_processes(duthost)
-            self.wait_lc_healthy_if_sup(duthost, duthosts)
+            self.wait_lc_healthy_if_sup(duthost, duthosts, localhost)
 
     def test_memory_exhaustion(self, duthosts, enum_rand_one_per_hwsku_hostname, localhost):
         duthost = duthosts[enum_rand_one_per_hwsku_hostname]
@@ -77,7 +77,7 @@ def test_memory_exhaustion(self, duthosts, enum_rand_one_per_hwsku_hostname, loc
                       'DUT {} did not startup'.format(hostname))
         # Wait until all critical processes are healthy.
         wait_critical_processes(duthost)
-        self.wait_lc_healthy_if_sup(duthost, duthosts)
+        self.wait_lc_healthy_if_sup(duthost, duthosts, localhost)
         # Verify DUT uptime is later than the time when the test case started running.
         dut_uptime = duthost.get_up_time()
         pytest_assert(dut_uptime > dut_datetime, "Device {} did not reboot".format(hostname))