lib/pragma/macro/model.rb in pragma-2.4.0 vs lib/pragma/macro/model.rb in pragma-2.5.0

@@ -4,12 +4,14 @@
 
 module Pragma
   module Macro
     def self.Model(action = nil)
       step = lambda do |input, options|
+        klass = Macro.require_skill('Model', 'model.class', options)
+
         Trailblazer::Operation::Pipetree::Step.new(
-          Trailblazer::Operation::Model.for(options['model.class'], action),
-          'model.class' => options['model.class'],
+          Trailblazer::Operation::Model.for(klass, action),
+          'model.class' => klass,
           'model.action' => action
         ).call(input, options).tap do |result|
           unless result
             options['result.response'] = Pragma::Operation::Response::NotFound.new.decorate_with(
               Pragma::Decorator::Error