lib/nanoc/base/compilation/rule_memory_calculator.rb in nanoc-4.0.0b3 vs lib/nanoc/base/compilation/rule_memory_calculator.rb in nanoc-4.0.0b4

@@ -17,13 +17,13 @@
     # @param [#reference] obj The object to calculate the rule memory for
     #
     # @return [Array] The caluclated rule memory for the given object
     def [](obj)
       result =
-        case obj.type
-        when :item_rep
+        case obj
+        when Nanoc::Int::ItemRep
           @rules_collection.new_rule_memory_for_rep(obj)
-        when :layout
+        when Nanoc::Int::Layout
           @rules_collection.new_rule_memory_for_layout(obj)
         else
           raise "Do not know how to calculate the rule memory for #{obj.inspect}"
         end