diff --git a/SKPhotoBrowser/extensions/UIView+ScreenshotProtection.swift b/SKPhotoBrowser/extensions/UIView+ScreenshotProtection.swift
index f3767653..46a3ac03 100644
--- a/SKPhotoBrowser/extensions/UIView+ScreenshotProtection.swift
+++ b/SKPhotoBrowser/extensions/UIView+ScreenshotProtection.swift
@@ -6,7 +6,7 @@
 //  Copyright © 2023 suzuki_keishi. All rights reserved.
 //
 
-import Foundation
+import UIKit
 
 extension UIView {
     func protectScreenshot() {