Fix ARC compile error