Update schema for flag update