Fix UIKit import