Updating to reflect UI change.