Fix for latest Swift/UIKit