diff --git a/lib/redlock/testing.rb b/lib/redlock/testing.rb
index 4de00ff..4524196 100644
--- a/lib/redlock/testing.rb
+++ b/lib/redlock/testing.rb
@@ -4,7 +4,7 @@ class Client
 
     alias_method :try_lock_instances_without_testing, :try_lock_instances
 
-    def try_lock_instances(resource, ttl)
+    def try_lock_instances(resource, ttl, extend)
       if @testing_mode == :bypass
         {
           validity: ttl,
@@ -14,7 +14,7 @@ def try_lock_instances(resource, ttl)
       elsif @testing_mode == :fail
         false
       else
-        try_lock_instances_without_testing resource, ttl
+        try_lock_instances_without_testing resource, ttl, extend
       end
     end