Update for new feature.