diff --git a/packages/selfie-segmentation/ios/Plugin/Classes/ProcessImageOptions.swift b/packages/selfie-segmentation/ios/Plugin/Classes/ProcessImageOptions.swift
index 41b55e9..a68f342 100644
--- a/packages/selfie-segmentation/ios/Plugin/Classes/ProcessImageOptions.swift
+++ b/packages/selfie-segmentation/ios/Plugin/Classes/ProcessImageOptions.swift
@@ -1,4 +1,5 @@
 import Foundation
+import UIKit
 
 extension UIImage {
     public func scaledImage(width: Int?, height: Int?) -> UIImage {