From f543e476522cc54bfc4911c0cc0dc731495ca77d Mon Sep 17 00:00:00 2001
From: Adel El-Beik <adel.el-beik@r3.com>
Date: Wed, 2 Aug 2023 13:39:04 +0100
Subject: [PATCH] ENT-10371: Fix unit test failure in
 CordaServiceLifecycleTests.

---
 .../net/corda/node/services/CordaServiceLifecycleTests.kt       | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/node/src/integration-test/kotlin/net/corda/node/services/CordaServiceLifecycleTests.kt b/node/src/integration-test/kotlin/net/corda/node/services/CordaServiceLifecycleTests.kt
index e615e91a60..93e0f7e10f 100644
--- a/node/src/integration-test/kotlin/net/corda/node/services/CordaServiceLifecycleTests.kt
+++ b/node/src/integration-test/kotlin/net/corda/node/services/CordaServiceLifecycleTests.kt
@@ -156,6 +156,6 @@ class CordaServiceLifecycleTests {
         /**
          * Given an event, was the SMM started when the event was received?
          */
-        fun getSmmStartedForEvent(event: ServiceLifecycleEvent) : Int = smmStateAtEvent.getOrDefault(event, STATE_MACHINE_MANAGER_UNKNOWN_STATUS)
+        fun getSmmStartedForEvent(event: ServiceLifecycleEvent) : Int  = smmStateAtEvent[event] ?: checkSmmStarted()
     }
 }
\ No newline at end of file