Update for Flags v2 API