lib/azeroth/model.rb in azeroth-0.7.0 vs lib/azeroth/model.rb in azeroth-0.7.1

@@ -60,10 +60,12 @@
     #
     # Returns options
     #
     # @return [Azeroth::Options]
 
+    delegate :decorator, to: :options
+
     # @private
     #
     # Returns decorator class for the object
     #
     # @return [Class] subclass of {Decorator}
@@ -83,10 +85,10 @@
     # returns DummyDecorator
     #
     # @return [Azeroth::Decorator,DummyDecorator]
     def calculate_decorator_class
       return DummyDecorator unless options.decorator
-      return options.decorator if options.decorator.is_a?(Class)
+      return decorator if decorator.is_a?(Class)
 
       klass::Decorator
     rescue NameError
       DummyDecorator
     end