Fixed missing include of UIKit