diff --git a/swift/StableDiffusionCLI/main.swift b/swift/StableDiffusionCLI/StableDiffusionSample.swift
similarity index 98%
rename from swift/StableDiffusionCLI/main.swift
rename to swift/StableDiffusionCLI/StableDiffusionSample.swift
index 4da8e62..decea05 100644
--- a/swift/StableDiffusionCLI/main.swift
+++ b/swift/StableDiffusionCLI/StableDiffusionSample.swift
@@ -9,6 +9,7 @@ import StableDiffusion
 import UniformTypeIdentifiers
 
 @available(iOS 16.2, macOS 13.1, *)
+@main
 struct StableDiffusionSample: ParsableCommand {
 
     static let configuration = CommandConfiguration(
@@ -205,9 +206,3 @@ enum SchedulerOption: String, ExpressibleByArgument {
         }
     }
 }
-
-if #available(iOS 16.2, macOS 13.1, *) {
-    StableDiffusionSample.main()
-} else {
-    print("Unsupported OS")
-}