diff --git a/lib/wings/setup.rb b/lib/wings/setup.rb
index e6b8bc697c..cebed2a6cb 100644
--- a/lib/wings/setup.rb
+++ b/lib/wings/setup.rb
@@ -128,8 +128,8 @@ def build_or_set(attributes, &block)
 end
 
 Wings::ModelRegistry.register(Hyrax::AccessControl,     Hydra::AccessControl)
-Wings::ModelRegistry.register(Hyrax::AdministrativeSet, AdminSet)
-Wings::ModelRegistry.register(Hyrax::PcdmCollection,    ::Collection)
+Wings::ModelRegistry.register(Hyrax.config.admin_set_class, AdminSet)
+Wings::ModelRegistry.register(Hyrax.config.collection_class, ::Collection)
 Wings::ModelRegistry.register(Hyrax::FileSet,           FileSet)
 Wings::ModelRegistry.register(Hyrax::Embargo,           Hydra::AccessControls::Embargo)
 Wings::ModelRegistry.register(Hyrax::Lease,             Hydra::AccessControls::Lease)