Update to remove old invalid class.