diff --git a/test/type_factory_test.rb b/test/type_factory_test.rb
index 3471e58cf..27bd7c302 100644
--- a/test/type_factory_test.rb
+++ b/test/type_factory_test.rb
@@ -135,7 +135,7 @@ def test_type
         assert_equal "?{ (Integer) -> void }", type.block.to_s
       end
 
-      factory.type(RBS::Types::Variable.new(name: :T, location: nil)) do |type|
+      factory.type(RBS::Types::Variable.new(name: :T, location: nil)).yield_self do |type|
         assert_instance_of Types::Var, type
         assert_equal :T, type.name
       end