Support multiple rewards in AIR