Fixing tests after major iOS commit

Just a quick fix to repair tests.