Make it work for ios