lib/simple_deploy/stack.rb in simple_deploy-0.7.4 vs lib/simple_deploy/stack.rb in simple_deploy-0.7.5

@@ -63,9 +63,10 @@
 
     def in_progress_update(args)
       if args[:caller].kind_of? Stack::Deployment::Status
         @logger.info "Updating #{@name}."
         attributes = stack_attribute_formatter.updated_attributes args[:attributes]
+        @template_body = args[:template_body] || template
 
         @entry.set_attributes attributes
         stack_updater.update_stack attributes
         @logger.info "Update complete for #{@name}."