Fixed missing UIKit include