diff --git a/Sources/AnyLanguageModel/Transcript.swift b/Sources/AnyLanguageModel/Transcript.swift
index 058935d..dcd1fad 100644
--- a/Sources/AnyLanguageModel/Transcript.swift
+++ b/Sources/AnyLanguageModel/Transcript.swift
@@ -456,7 +456,7 @@ extension Transcript.ImageSegment {
     }
 }
 
-#if canImport(UIKit)
+#if canImport(UIKit) || targetEnvironment(macCatalyst)
     import UIKit
 
     extension Transcript.ImageSegment {
@@ -486,9 +486,7 @@ extension Transcript.ImageSegment {
             self.init(data: data, mimeType: mimeType)
         }
     }
-#endif
-
-#if canImport(AppKit)
+#elseif canImport(AppKit)
     import AppKit
 
     extension Transcript.ImageSegment {