diff --git a/t/integration_block_req_header.t b/t/integration_block_req_header.t
index d505380..f526151 100644
--- a/t/integration_block_req_header.t
+++ b/t/integration_block_req_header.t
@@ -7,8 +7,8 @@ our $HttpConfig = <<'_EOC_';
     lua_need_request_body on;
     init_worker_by_lua_block{
             local coraza = require "resty.coraza"
-            coraza.do_init()
-            coraza.rules_add([[SecRule REQUEST_HEADERS:User-Agent "Mozilla" "phase:1, id:3,drop,status:452,log,msg:'Blocked User-Agent'"]])
+            waf = coraza.do_init()
+            coraza.rules_add(waf, [[SecRule REQUEST_HEADERS:User-Agent "Mozilla" "phase:1, id:3,drop,status:452,log,msg:'Blocked User-Agent'"]])
     }
 _EOC_
 
@@ -16,7 +16,7 @@ our $LocationConfig = <<'_EOC_';
     location /t {
             access_by_lua_block {
             local coraza = require "resty.coraza"
-            coraza.do_access_filter()
+            coraza.do_access_filter(waf)
             coraza.do_interrupt()
         }
 
diff --git a/t/integration_block_resp_header.t b/t/integration_block_resp_header.t
index db119ae..658852d 100644
--- a/t/integration_block_resp_header.t
+++ b/t/integration_block_resp_header.t
@@ -7,8 +7,8 @@ our $HttpConfig = <<'_EOC_';
     lua_need_request_body on;
     init_worker_by_lua_block{
             local coraza = require "resty.coraza"
-            coraza.do_init()
-            coraza.rules_add([[SecRule RESPONSE_HEADERS:Content-Type "text" "phase:3, id:4,drop,status:451,log,msg:'Blocked content-type'"]])
+            waf = coraza.do_init()
+            coraza.rules_add(waf, [[SecRule RESPONSE_HEADERS:Content-Type "text" "phase:3, id:4,drop,status:451,log,msg:'Blocked content-type'"]])
     }
 _EOC_
 
@@ -16,7 +16,7 @@ our $LocationConfig = <<'_EOC_';
     location /t {
             access_by_lua_block {
             local coraza = require "resty.coraza"
-            coraza.do_access_filter()
+            coraza.do_access_filter(waf)
             coraza.do_interrupt()
         }
 
diff --git a/t/integration_passed.t b/t/integration_passed.t
index a627d46..7311e07 100644
--- a/t/integration_passed.t
+++ b/t/integration_passed.t
@@ -7,7 +7,7 @@ our $HttpConfig = <<'_EOC_';
     lua_need_request_body on;
     init_worker_by_lua_block{
             local coraza = require "resty.coraza"
-            coraza.do_init()
+            waf = coraza.do_init()
    }
 _EOC_
 
@@ -15,7 +15,7 @@ our $LocationConfig = <<'_EOC_';
     location /t {
             access_by_lua_block {
             local coraza = require "resty.coraza"
-            coraza.do_access_filter()
+            coraza.do_access_filter(waf)
             coraza.do_interrupt()
         }
 
diff --git a/t/integration_with_coreruleset.t b/t/integration_with_coreruleset.t
index 9e7f63c..1871e8e 100644
--- a/t/integration_with_coreruleset.t
+++ b/t/integration_with_coreruleset.t
@@ -7,9 +7,9 @@ our $HttpConfig = <<'_EOC_';
     lua_need_request_body on;
     init_worker_by_lua_block{
             local coraza = require "resty.coraza"
-            coraza.do_init()
-            coraza.rules_add_file("%s/t/coraza.conf")
-            coraza.rules_add("Include %s/t/coreruleset/rules/*.conf")
+            waf = coraza.do_init()
+            coraza.rules_add_file(waf, "%s/t/coraza.conf")
+            coraza.rules_add(waf, "Include %s/t/coreruleset/rules/*.conf")
     }
 _EOC_
 
@@ -19,7 +19,7 @@ our $LocationConfig = <<'_EOC_';
     location /t {
             access_by_lua_block {
             local coraza = require "resty.coraza"
-            coraza.do_access_filter()
+            coraza.do_access_filter(waf)
             coraza.do_interrupt()
         }