diff --git a/server-tools/persona-compute-pool/test_scheduler.py b/server-tools/persona-compute-pool/test_scheduler.py
index 56c9356..aa88971 100644
--- a/server-tools/persona-compute-pool/test_scheduler.py
+++ b/server-tools/persona-compute-pool/test_scheduler.py
@@ -9,8 +9,12 @@ class SchedulerTests(unittest.TestCase):
     def test_audit_excludes_mother_node(self):
         value=M.audit(); self.assertEqual(value["outcome"],"PASS"); self.assertTrue(value["mother_node_excluded"])
     def test_pending_workers_do_not_fake_a_lease(self):
-        value=M.select({"min_cpu_cores":1,"min_memory_mb":256}); self.assertEqual(value["outcome"],"NO_ELIGIBLE_WORKER")
+        pool=M.load_map()
+        for node in pool["nodes"]: node["state"]="DISCOVERED_OPT_IN_PENDING_AGENT"
+        value=M.select({"min_cpu_cores":1,"min_memory_mb":256},pool); self.assertEqual(value["outcome"],"NO_ELIGIBLE_WORKER")
     def test_active_worker_is_selected_by_capacity(self):
-        pool=M.load_map(); pool["nodes"][2]["state"]="ACTIVE_WORKER"; value=M.select({"min_cpu_cores":2,"min_memory_mb":1000},pool); self.assertEqual(value["outcome"],"LEASE_REQUESTED"); self.assertEqual(value["node_id"],"XX-GZ-001")
+        pool=M.load_map()
+        for node in pool["nodes"]: node["state"]="DISCOVERED_OPT_IN_PENDING_AGENT"
+        pool["nodes"][2]["state"]="ACTIVE_WORKER"; value=M.select({"min_cpu_cores":2,"min_memory_mb":1000},pool); self.assertEqual(value["outcome"],"LEASE_REQUESTED"); self.assertEqual(value["node_id"],"XX-GZ-001")
 
 if __name__=="__main__": unittest.main()