lib/appfuel/storage/repository/base.rb in appfuel-0.4.5 vs lib/appfuel/storage/repository/base.rb in appfuel-0.5.0

@@ -193,12 +193,12 @@
 
       def to_storage(entity, exclude = [])
         mapper.to_storage(entity, exclude)
       end
 
-      def to_entity(domain_name, storage)
+      def to_entity(domain_name, type, storage)
         key  = qualify_container_key(domain_name, "domains")
-        hash = mapper.to_entity_hash(domain_name, storage)
+        hash = mapper.to_entity_hash(domain_name, type, storage)
         app_container[key].new(hash)
       end
 
       def build(type:, name:, storage:, **inputs)
         builder = find_entity_builder(type: type, domain_name: name)