lib/cucumber/pickles/transform.rb in pickles-0.1.4 vs lib/cucumber/pickles/transform.rb in pickles-0.1.5

@@ -1,15 +1,18 @@
 within_reg = /\A\s*(.*)?\s*(?:["|'](.*?)["|'])?\s*\Z/
 
 Transform(/(within .*)$/) do |within_info|
   splitted = within_info.split('within').reject(&:blank?)
 
-  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)
+    Waiter.wait do
+      within = Pickles.detect_node(el_alias, locator, within: within)
+    end
+
+    within
   end
 end