Adjusted to work with Swift in Xcode 8 GM

Quick fix to make this compile in the latest Xcode