diff --git a/Sources/coreml-stable-diffusion-swift/helper/HelperImage.swift b/Sources/coreml-stable-diffusion-swift/helper/HelperImage.swift
index 78c4725..0f71236 100644
--- a/Sources/coreml-stable-diffusion-swift/helper/HelperImage.swift
+++ b/Sources/coreml-stable-diffusion-swift/helper/HelperImage.swift
@@ -18,6 +18,7 @@ import AppKit
 ///   - toSize: Fit to size
 /// - Returns: NSImage
 @available(macOS 13.1, *)
+@MainActor
 public func getNSImage(from data : Data?, cropped toSize : NSSize? = nil) async -> NSImage? {
 
     guard let value = data else{