Fix code to match with Beta5 build of Swift.