Updated for new flag