Update for Swift 1.1