lib/floe/workflow/states/wait.rb in floe-0.9.0 vs lib/floe/workflow/states/wait.rb in floe-0.10.0

@@ -27,16 +27,21 @@
         end
 
         def start(input)
           super
 
-          input          = input_path.value(context, input)
-          context.output = output_path.value(context, input)
+          input = input_path.value(context, context.input)
 
           wait_until!(
             :seconds => seconds_path ? seconds_path.value(context, input).to_i : seconds,
             :time    => timestamp_path ? timestamp_path.value(context, input) : timestamp
           )
+        end
+
+        def finish
+          input          = input_path.value(context, context.input)
+          context.output = output_path.value(context, input)
+          super
         end
 
         def running?
           waiting?
         end