Import UIKit to prevent build error