Update for new Swift version