Fix for updated class name.