Update hook_flag to 3.x API.