Set the correct class name.