Update to Flags v2 API