diff --git a/Source/Device.generated.swift b/Source/Device.generated.swift
index 77186f7..f492b0b 100644
--- a/Source/Device.generated.swift
+++ b/Source/Device.generated.swift
@@ -11,8 +11,10 @@
 
 #if os(watchOS)
 import WatchKit
-#else
+#elseif canImport(UIKit)
 import UIKit
+#else
+import Foundation
 #endif
 
 // MARK: Device
diff --git a/Source/Device.swift.gyb b/Source/Device.swift.gyb
index d99c280..8b0fa98 100644
--- a/Source/Device.swift.gyb
+++ b/Source/Device.swift.gyb
@@ -299,8 +299,10 @@ watchOSDevices = watches
 }%
 #if os(watchOS)
 import WatchKit
-#else
+#elseif canImport(UIKit)
 import UIKit
+#else
+import Foundation
 #endif
 
 // MARK: Device