diff --git a/Sources/IoT-Simulator-Core/IoT_Simulator_Core.swift b/Sources/IoT-Simulator-Core/IoT_Simulator_Core.swift
index b1476bd..6dbb6a6 100644
--- a/Sources/IoT-Simulator-Core/IoT_Simulator_Core.swift
+++ b/Sources/IoT-Simulator-Core/IoT_Simulator_Core.swift
@@ -101,7 +101,7 @@ public actor IoTSimulatorCore {
     private static func schedule(
         envID: sending String,
         deviceID: UInt128,
-        success: sending @escaping (_ msg: Data) throws -> Void,
+        success: sending @escaping (_ msg: Data) async throws -> Void,
         failure: sending @escaping () -> Void
     ) -> Task<(), Never>{
         let _devID: String = "\(deviceID)"
@@ -117,7 +117,7 @@ public actor IoTSimulatorCore {
                 do {
                     let message = try dev.work(envrionment: env)
 
-                    try success(message)
+                    try await success(message)
                 } catch {
                     failure()
                 }