diff --git a/test/integration_test.rb b/test/integration_test.rb
index 1bcba44..435a2fd 100644
--- a/test/integration_test.rb
+++ b/test/integration_test.rb
@@ -9,7 +9,7 @@
 
   after do
     # Reset Circuitbox
-    if Circuitbox.method_defined?(:reset)
+    if Circuitbox.respond_to?(:reset)
       Circuitbox.reset
     else
       Circuitbox.configure { |config| config.default_circuit_store = Circuitbox::MemoryStore.new }
diff --git a/test/vertex_client_test.rb b/test/vertex_client_test.rb
index 4228214..3769348 100644
--- a/test/vertex_client_test.rb
+++ b/test/vertex_client_test.rb
@@ -44,7 +44,7 @@
 
   describe 'circuit' do
     before do
-      if Circuitbox.method_defined?(:reset)
+      if Circuitbox.respond_to?(:reset)
         Circuitbox.reset
       else
         Circuitbox.configure { |config| config.default_circuit_store = Circuitbox::MemoryStore.new }
@@ -54,7 +54,7 @@
     end
 
     after do
-      if Circuitbox.method_defined?(:reset)
+      if Circuitbox.respond_to?(:reset)
         Circuitbox.reset
       else
         Circuitbox.configure { |config| config.default_circuit_store = Circuitbox::MemoryStore.new }