lib/ajax/spec/helpers.rb in ajax-0.1.4 vs lib/ajax/spec/helpers.rb in ajax-0.1.5

@@ -28,12 +28,12 @@
       def should_set_ajax_response_header(key, value)
         response_headers['Ajax-Info'][key].should == value
       end
 
       def should_set_ajax_request_header(key, value)
-        @env['Ajax-Info'][key].should == value
+        Ajax.get_header(@env, key).should == value
       end
-            
+
       def should_rewrite_to(url)
         should_be_a_valid_response
 
         # Check custom headers
         response_body_as_hash['REQUEST_URI'].should == url
\ No newline at end of file