Correct bug on new feature.