Add missing ClassRegistry