Fix UIKit dependency