Fix ARC issue