diff --git a/integration-tests/req-set-header.rb b/integration-tests/req-set-header.rb
index 6f63e357..27e9dc34 100644
--- a/integration-tests/req-set-header.rb
+++ b/integration-tests/req-set-header.rb
@@ -1,3 +1,3 @@
-Nghttpx.run do |resp, req|
-  req.set_header "User-Agent", "mruby"
+Nghttpx.run do |env|
+  env.req.set_header "User-Agent", "mruby"
 end
diff --git a/integration-tests/resp-set-header.rb b/integration-tests/resp-set-header.rb
index a1f2d806..b947ce3b 100644
--- a/integration-tests/resp-set-header.rb
+++ b/integration-tests/resp-set-header.rb
@@ -1,3 +1,3 @@
-Nghttpx.run do |resp, req|
-  resp.set_header "Alpha", "bravo"
+Nghttpx.run do |env|
+  env.resp.set_header "Alpha", "bravo"
 end
diff --git a/integration-tests/return.rb b/integration-tests/return.rb
index 42b1e345..907d837f 100644
--- a/integration-tests/return.rb
+++ b/integration-tests/return.rb
@@ -1,4 +1,6 @@
-Nghttpx.run do |resp, req|
+Nghttpx.run do |env|
+  resp = env.resp
+
   resp.clear_headers
   resp.status = 404
   resp.add_header "from", "mruby"