lib/appfuel/domain/dsl.rb in appfuel-0.2.3 vs lib/appfuel/domain/dsl.rb in appfuel-0.2.4

@@ -25,11 +25,15 @@
         klass.instance_variable_set(:@value_object, false)
         klass.instance_variable_set(:@schema, {})
         klass.equalizer = Dry::Equalizer.new(*schema.keys)
         klass.send(:include, klass.equalizer)
 
-        register_container_class(klass)
+        stage_class_for_registration(klass)
         Types.register_domain(klass)
+      end
+
+      def container_class_type
+        'domains'
       end
 
       def default?
         false
       end