making some quick fix to run in Xcode