lib/cucumber/pickles/transform.rb in pickles-0.1.2 vs lib/cucumber/pickles/transform.rb in pickles-0.1.3

@@ -1,10 +1,12 @@
 within_reg = /\A\s*(.*)?\s*(?:["|'](.*?)["|'])?\s*\Z/
 
 Transform(/(within .*)$/) do |within_info|
   splitted = within_info.split('within').reject(&:blank?)
 
-  splitted.reverse_each.each_with_object(page) do |info, within|
+  Waiter.wait_for_ajax
+
+  splitted.reverse_each.inject(page) do |within, info|
     captures = Helpers::Regex::WITHIN.match(info).captures
     el_alias = captures[0]
     locator = captures[1]
 
     Pickles.detect_node(el_alias, locator, within: within)