diff --git a/lldb/test/API/tools/lldb-server/TestGdbRemoteForkNonStop.py b/lldb/test/API/tools/lldb-server/TestGdbRemoteForkNonStop.py
index 6b5010da9b1934..bd9484ff0903e6 100644
--- a/lldb/test/API/tools/lldb-server/TestGdbRemoteForkNonStop.py
+++ b/lldb/test/API/tools/lldb-server/TestGdbRemoteForkNonStop.py
@@ -92,8 +92,7 @@ def test_kill_all_nonstop(self):
             "send packet: $OK#00",
         ], True)
         self.expect_gdbremote_sequence()
-        self.assertEqual(set([ret["pid1"], ret["pid2"]]),
-                         set([parent_pid, child_pid]))
+        self.assertEqual(set([pid1, pid2]), set([parent_pid, child_pid]))
 
     @add_test_categories(["fork"])
     def test_vkill_both_nonstop(self):