Fix compilation errors

After PR #166 was merged, I synced my code and tried to compile a new version. I was all set but then I got two compilation errors in Xcode. Simple fix, though.