diff --git a/test/test_instrumentation.rb b/test/test_instrumentation.rb
index b9e5c3e9..9fd5dc66 100644
--- a/test/test_instrumentation.rb
+++ b/test/test_instrumentation.rb
@@ -14,7 +14,7 @@ class TestInstrumentation < Minitest::Test
 
     def instrument(name, *args)
       @events << Event.new(name, args)
-      yield
+      yield if block_given?
     end
   end