lib/elasticity/job_flow.rb in elasticity-2.5.1 vs lib/elasticity/job_flow.rb in elasticity-2.5.2

@@ -96,11 +96,11 @@
 
     def add_step(jobflow_step)
       if is_jobflow_running?
         jobflow_steps = []
         if jobflow_step.requires_installation? && !@installed_steps.include?(jobflow_step.class)
-          jobflow_steps << jobflow_step.aws_installation_step
+          jobflow_steps.concat(jobflow_step.aws_installation_steps)
         end
         jobflow_steps << jobflow_step.to_aws_step(self)
         emr.add_jobflow_steps(@jobflow_id, {:steps => jobflow_steps})
       else
         @jobflow_steps << jobflow_step
@@ -170,10 +170,10 @@
 
     def jobflow_steps
       steps = []
       @jobflow_steps.each do |step|
         if step.class.send(:requires_installation?) && !@installed_steps.include?(step.class)
-          steps << step.class.send(:aws_installation_step)
+          steps.concat(step.class.send(:aws_installation_steps))
           @installed_steps << step.class
         end
         steps << step.to_aws_step(self)
       end
       steps
\ No newline at end of file