diff --git a/swift/ThumbHash.swift b/swift/ThumbHash.swift
index 8e461f1..3a399b5 100644
--- a/swift/ThumbHash.swift
+++ b/swift/ThumbHash.swift
@@ -555,7 +555,7 @@ func thumbHashToImage(hash: Data) -> NSImage {
 }
 #endif
 
-#if os(iOS)
+#if os(iOS) || os(tvOS)
 import UIKit
 
 func imageToThumbHash(image: UIImage) -> Data {